File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -25,11 +25,11 @@ multi_scale_infer=False
25
25
shuffle_AB=False
26
26
27
27
# Initializing from pretrained weights
28
- pretrain=/media/lidan/ssd2/ChangeFormer/pretrained_segformer/segformer.b2.512x512.ade.160k.pth
28
+ pretrain=pretrained_changeformer/pretrained_changeformer.pt
29
29
30
30
# Train and Validation splits
31
31
split=train # trainval
32
32
split_val=test # test
33
33
project_name=CD_${net_G} _${data_name} _b${batch_size} _lr${lr} _${optimizer} _${split} _${split_val} _${max_epochs} _${lr_policy} _${loss} _multi_train_${multi_scale_train} _multi_infer_${multi_scale_infer} _shuffle_AB_${shuffle_AB} _embed_dim_${embed_dim}
34
34
35
- CUDA_VISIBLE_DEVICES=1 python main_cd.py --img_size ${img_size} --loss ${loss} --checkpoint_root ${checkpoint_root} --vis_root ${vis_root} --lr_policy ${lr_policy} --optimizer ${optimizer} --pretrain ${pretrain} --split ${split} --split_val ${split_val} --net_G ${net_G} --multi_scale_train ${multi_scale_train} --multi_scale_infer ${multi_scale_infer} --gpu_ids ${gpus} --max_epochs ${max_epochs} --project_name ${project_name} --batch_size ${batch_size} --shuffle_AB ${shuffle_AB} --data_name ${data_name} --lr ${lr} --embed_dim ${embed_dim}
35
+ CUDA_VISIBLE_DEVICES=1 python main_cd.py --img_size ${img_size} --loss ${loss} --checkpoint_root ${checkpoint_root} --vis_root ${vis_root} --lr_policy ${lr_policy} --optimizer ${optimizer} --pretrain ${pretrain} --split ${split} --split_val ${split_val} --net_G ${net_G} --multi_scale_train ${multi_scale_train} --multi_scale_infer ${multi_scale_infer} --gpu_ids ${gpus} --max_epochs ${max_epochs} --project_name ${project_name} --batch_size ${batch_size} --shuffle_AB ${shuffle_AB} --data_name ${data_name} --lr ${lr} --embed_dim ${embed_dim}
You can’t perform that action at this time.
0 commit comments