pspnet-pytorch icon indicating copy to clipboard operation
pspnet-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

pspnet-resnet50.pth

Open zhijiejia opened this issue 4 years ago • 4 comments

请问下载的pspnet_resnet50.pth是作者您已经训练好的是吗？ 我在使用train.py 训练时，使用了Resnet50的预训练模型，是不是需要把加载pspnet_resnet50.pth的代码注释掉呢？

zhijiejia avatar Dec 20 '20 10:12 zhijiejia

image 应该是吧，我这一个epoch都没跑完，loss就只有0.121了，哈哈哈

zhijiejia avatar Dec 20 '20 10:12 zhijiejia

是的

bubbliiiing avatar Dec 22 '20 07:12 bubbliiiing

你好，我参考了你的代码，仅使用了预训练的resnet50，没有使用预训练的pspnet模型，在voc_aug上按照默认设置训练，mIoU仅能达到72%左右，请问一下，训练到82%对于超参需要怎样设置呀？

wywd avatar Sep 16 '21 09:09 wywd

你学习率调大一些试试，前面是5e-4后面是1e-4

bubbliiiing avatar Sep 19 '21 15:09 bubbliiiing