bubbliiiing/unet-pytorch

我的cpu直接占了100%

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