解决方法:升级 TorchVision
问题原因2:TorchVision 版本过高
解决方法:将
from torchvision.models.utils import load_state_dict_from_url更改为
from torch.hub import load_state_dict_from_url
欢迎分享,转载请注明来源:内存溢出
解决方法:升级 TorchVision
问题原因2:TorchVision 版本过高
解决方法:将
from torchvision.models.utils import load_state_dict_from_url更改为
from torch.hub import load_state_dict_from_url
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)