ProDA icon indicating copy to clipboard operation
ProDA copied to clipboard

Exact command line of warmup stage

Open shahaf1313 opened this issue 2 years ago • 3 comments

Hey :) Can you guys please supply the command line used to train the warmup phase? I'm trying to retrain a network "from the beginning". Thanks! Shahaf

shahaf1313 avatar Dec 24 '21 11:12 shahaf1313

I have similar question.

The provided checkpoint (warm-up model) is 43.3 mIoU, while the reported metric in the paper is 41.6 mIoU, which quite confused me.

pascal1129 avatar Feb 24 '22 08:02 pascal1129

You can try one of below commands to get warmup model:

python train.py --name gta2cityv2_warmupd_ls --model_name deeplabv2 --warm_up --freeze_bn --gan LS --lr 2.5e-4 --adv 0.01 --no_resume

python train.py --name gta2cityv2_warmupd_ls_S --model_name deeplabv2 --warm_up --freeze_bn --gan LS --lr 2.5e-4 --adv 0.01 --no_resume --S_pseudo_src 1

panzhang0104 avatar Apr 01 '22 03:04 panzhang0104

@pascal1129 In order to make sure the released code can be reproduced, we clean the code and retrain the whole framework. So the released model is from the clean code. Since the warmup stage is trained by adversarial loss, it is unstable. The result of this stage maybe a little bit different but does not influence the result of the final stage.

panzhang0104 avatar Apr 01 '22 03:04 panzhang0104