ooooverflow / BiSeNet

BiSeNet based on pytorch
394 stars 77 forks source link

Start counting epochs from this number wrong #6

Open GritCoder opened 5 years ago

GritCoder commented 5 years ago

train.py里面并没有实现epoch_start_i函数判断的模块,所以当引用这个参数会报错。 因此目前来说,如果想接着上次迭代的epoch来进行训练的话,只能通过pre-train model 来操作

zz10001 commented 5 years ago

我中断了程序,然后重新开始重中断的地方开始跑应该怎么做。就(假如从第三十次中断)在train.py里面的params加'--pretrained_model_path','checkpoints_101_adam/epoch_30.pth'这样吗。但他显示还是epoch1而不是epoch31。

Jingchensun commented 5 years ago

我中断了程序,然后重新开始重中断的地方开始跑应该怎么做。就(假如从第三十次中断)在train.py里面的params加'--pretrained_model_path','checkpoints_101_adam/epoch_30.pth'这样吗。但他显示还是epoch1而不是epoch31。

hallo想问下你最开始用的预训练模型是哪里来的,可以发我一个链接吗

zz10001 commented 5 years ago

hallo想问下你最开始用的预训练模型是哪里来的,可以发我一个链接吗

不是跑的时候直接下载的resnet18和resnet101吗。你说的预训练模型什么意思。

zz10001 commented 5 years ago

hallo想问下你最开始用的预训练模型是哪里来的,可以发我一个链接吗

epoch295那个模型吗,我是自己训练出来的,不过感觉这个网络只能跑到acc:0.93 miou:0.58(train)就上不去了。

Jingchensun commented 5 years ago

hallo想问下你最开始用的预训练模型是哪里来的,可以发我一个链接吗

不是跑的时候直接下载的resnet18和resnet101吗。你说的预训练模型什么意思。

对我指的就是epoch295那个模型,它给的那个模型部署有问题吗。你说的resenet18和resenet101是直接通过pytorch库下载的吗,我在train.py文件里没有看到下载的程序呀

Jingchensun commented 5 years ago

hallo想问下你最开始用的预训练模型是哪里来的,可以发我一个链接吗

epoch295那个模型吗,我是自己训练出来的,不过感觉这个网络只能跑到acc:0.93 miou:0.58(train)就上不去了。

可以看下这个精度排名https://paperswithcode.com/sota/real-time-semantic-segmentation-on-camvid 另外,有没有测试过其他速度如何?cityscape数据集上跑一下看

zz10001 commented 5 years ago

可以看下这个精度排名https://paperswithcode.com/sota/real-time-semantic-segmentation-on-camvid 另外,有没有测试过其他速度如何?cityscape数据集上跑一下看

他这个排名应该不全,你可以先自己训练。然后用保存的模型测试一下应该不会报错。我是自己训练出来的,然后去测试的没有问题 image

Jingchensun commented 5 years ago

可以看下这个精度排名https://paperswithcode.com/sota/real-time-semantic-segmentation-on-camvid 另外,有没有测试过其他速度如何?cityscape数据集上跑一下看

他这个排名应该不全,你可以先自己训练。然后用保存的模型测试一下应该不会报错。我是自己训练出来的,然后去测试的没有问题 image

好的解决了多谢~是我pytorch版本问题,之前是0.4.0,更新到0.4.1就可以正常下载了。

miscedence12 commented 2 years ago

@Jingchensun 你好,当我指定pretrained_model_path时再训练,报了以下错误: model.module.load_state_dict(torch.load(args.pretrained_model_path)) raise AttributeError("'{}' object has no attribute '{}'".format(AttributeError: 'BiSeNet' object has no attribute 'module'