bubbliiiing / unet-pytorch

这是一个unet-pytorch的源码,可以训练自己的模型

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

我的cpu直接占了100%

yqc1212443 opened this issue · 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啥的。
image