search

brandleyzhou / DIFFNet

[BMVC 2021] ''Self-Supervised Monocular Depth Estimation with Internal Feature Fusion''
109 stars 21 forks source link

Training #24

Open ljy199712 opened 1 year ago

ljy199712 commented 1 year ago

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?

brandleyzhou commented 1 year ago

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

ljy199712 commented 1 year ago

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

brandleyzhou commented 1 year ago

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