Map-Detect / scripts /run_cd.sh
justchenhao's picture
BIT
d43892c
Raw
History Blame Contribute Delete
696 Bytes
#!/usr/bin/env bash
gpus=0
checkpoint_root=checkpoints
data_name=LEVIR
img_size=256
batch_size=8
lr=0.01
max_epochs=200
net_G=base_transformer_pos_s4_dd8
#base_resnet18
#base_transformer_pos_s4_dd8
#base_transformer_pos_s4_dd8_dedim8
lr_policy=linear
split=trainval
split_val=test
project_name=CD_${net_G}_${data_name}_b${batch_size}_lr${lr}_${split}_${split_val}_${max_epochs}_${lr_policy}
python main_cd.py --img_size ${img_size} --checkpoint_root ${checkpoint_root} --lr_policy ${lr_policy} --split ${split} --split_val ${split_val} --net_G ${net_G} --gpu_ids ${gpus} --max_epochs ${max_epochs} --project_name ${project_name} --batch_size ${batch_size} --data_name ${data_name} --lr ${lr}