Thea
Thea
Have you solved the problem? I encountered the same problem as you.
> 感谢您的回复。我的理解是对代码auto_load_resume.py中第18到21行，删掉model_state_dict中的module的作用是因为使用多GPU训练导致加载模型出现问题。如果是仅使用单卡训练，将其注释掉或者加上判断module是否存在再进行删除解决了报错问题。可以这样理解吗？ 我训练的时候也遇到过这种情况，自己根据错误调整一下就行了，后面我尝试改成多卡训练，会报一个没见过也不知道怎么解决的错误，使用想问问你实现了多卡训练吗？ 实现了的话，可否告知一下，如何实现的呀？
目前我跑的精度分别为，cub：89.25%，aircraft : 94.06%，cars：94.75%。 想知道如何调整才能达到你论文的精度呢？ 期待您的回复，谢谢！
> 谢谢！主要是augmentation添加了color distort > https://github.com/ZF1044404254/MMAL-Net/blob/199c1706695ba0f645809d3568881816fbe2fb70/datasets/dataset.py#L62 哇喔 回的好快 受宠若惊 知道啦 我去改一改试试 非常感谢！！！