运行MixVPR项目中遇到的问题

这篇文章记录在AutoDL上跑MixVPR项目遇到的问题。如果对于AutoDL的使用或者对于PyCharm将MixVPR远程部署到AutoDL云服务器有疑问的朋友,可以参看我的上一篇笔记AutoDL教程和MixVPR环境部署

1.路径问题

在这里插入图片描述
问题:原本路径应该是root/autodl-tmp/dataloaders/train/GSVCitiesDataset/Dataframes/Bangkok.csv 但是代码识别为root/autodl-tmp/dataloaders/train/GSVCitiesDatasetDataframes/Bangkok.csv 找到出错的文件
在这里插入图片描述
应该修改成下面的:
在这里插入图片描述
注意下面还有一个需要修改的路径,按照上面的方法就可以啦!

2.参数问题

在这里插入图片描述
报错信息:TypeError: _init() got an unexpected keyword argument 'weights! 从错误日志来看,问题出在你的代码试图调用 torchvision.models.resnet50 函数,并传给它一个名为 ‘weights’ 的参数,但这个函数并不接受这个参数,导致 Python 抛出 TypeError。
torchvision.models.resnet50 的正确调用方式如下:

torchvision.models.resnet50(pretrained=False, **kwargs)

其中,‘pretrained’ 如果为 True,将返回在 ImageNet 上预训练的模型;否则,将返回未初始化的模型。‘**kwargs’ 用于传递给 ResNet 类的其他参数。因此,对于代码,我们应该修改 “/root/autodl-tmp/models/backbones/resnet.py” 文件中的调用方式。将其中的

self.model = torchvision.models.resnet50(weights=weights)

改为

self.model = torchvision.models.resnet50(pretrained=weights)

在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值