yqc1212443 opened this issue a year ago · 0 comments
大佬你好,我感觉gpu的cuda也有60%,我改了vgg.py。是不是把from torchvision.models.utils import load_state_dict_from_url 改为了from torch.hub import load_state_dict_from_url的原因,要不然不改一直提醒我缺一个torch啥的。