这是一个unet-pytorch的源码,可以训练自己的模型
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
yqc1212443 opened this issue a year ago · 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啥的。