DIFFNet icon indicating copy to clipboard operation
DIFFNet copied to clipboard

Published 20 hours ago • verified publisher icon brandleyzhou

Training

Open ljy199712 opened this issue 2 years ago • 3 comments

Thanks for your working. Here are something detials i want 2 ask you . Here are my torch torch 1.7.1+cu110 torchaudio 0.7.2 torchsummary 1.5.1
torchvision 0.8.2+cu110 I found when i set the initial learning rate as 10−4 for the first 14 epochs and then 10−5 for last 5 epochs ,my experimental results are very different from yours . Is it the reason for different PyTorch versions？Or my training process wrong?

ljy199712 avatar Aug 28 '22 07:08 ljy199712

you are right, the PyTorch version effects. please use torch 1.1.0 for training

brandleyzhou avatar Aug 28 '22 09:08 brandleyzhou

Thanks for your time, a error in loading weight when i change the PyTorch versions. 微信图片_20220829102246

ljy199712 avatar Aug 29 '22 02:08 ljy199712

It seems you need to manually download the pretrained hrnet18. please refer to this issue https://github.com/brandleyzhou/DIFFNet/issues/7

brandleyzhou avatar Aug 29 '22 16:08 brandleyzhou