首先下载数据集和info文件;
报错:FileNotFoundError: [Errno 2] No such file or directory: '/home/ubuntu/.torch/models/resnet50-19c8e357.pth'
下载'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth';
在model.ResNet_hypergrapysage_part.py的543行修改resnet50-19c8e357.pth的地址为自己的地址,问题解决;
第二个问题:RuntimeError: CUDA out of memory.
修改train-batch:main_video_person_reid_hypergraphsage_part.py44行的32改为1,和run.sh的train_batch的32改为1
但又有问题ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 2048])
于是将train_batch改为8,解决
复现MGH多粒度超图REID-CVPR2020步骤
最新推荐文章于 2022-07-27 16:34:25 发布