DSRL icon indicating copy to clipboard operation
DSRL copied to clipboard

I can't seem to trained the model to get good result

Open gnsci opened this issue 3 years ago • 2 comments

Hi there, Thanks for sharing your codes. I wonder if the experience setting in your current train.py can achieve mIoU=0.6768 on Cityscapes. I trained the model, but I can get good result. Can you share the setting for getting mIoU=0.6768?

gnsci avatar Dec 06 '21 14:12 gnsci

Hi there, Thanks for sharing your codes. I wonder if the experience setting in your current train.py can achieve mIoU=0.6768 on Cityscapes. I trained the model, but I can get good result. Can you share the setting for getting mIoU=0.6768?

hi @gnsci i think maybe u can try start the training with lr=0.01, then load the weights and restart the training with a lower lr (say 0.001). By the way, the pretrained weights is provided in this repo, u can download it from google drive and directly load it into the network. however, note that this is NOT the official implementation of DSRL so u may find the result is not so good as what is presented in the paper.

Dootmaan avatar Dec 13 '21 04:12 Dootmaan

Thanks a lot!

gnsci avatar Dec 13 '21 04:12 gnsci