使用DIS5K数据集单GPU训练报错

[huchunjun425]

您好，请问一下，我使用单GPU训练github下载得指定代码，使用“./train_test.sh train 1 1”命令尝试训练代码时，报错如下：Single-GPU mode received... Traceback (most recent call last): File "train.py", line 12, in from models.birefnet import BiRefNet File "/mnt/data/code/semanticSegmentation/huChunJun/BiRefNet/models/birefnet.py", line 9, in from models.backbones.build_backbone import build_backbone File "/mnt/data/code/semanticSegmentation/huChunJun/BiRefNet/models/backbones/build_backbone.py", line 4, in from torchvision.models import vgg16, vgg16_bn, VGG16_Weights, VGG16_BN_Weights, resnet50, ResNet50_Weights ImportError: cannot import name 'VGG16_Weights' from 'torchvision.models' (/root/anaconda3/envs/hcjcnn/lib/python3.7/site-packages/torchvision/models/init.py) 请问一下这个是由于什么原因导致得？我按照要求进入build_backbone.py文件里面，发现from torchvision.models import vVGG16_Weights, VGG16_BN_Weights, ResNet50_Weights三个模块无法追踪。