使用pytorch神经网络时报错TypeError: __init__() takes 1 positional argument but 3 were given
项目场景:试图加入一个新的loss函数问题描述:在加入新的loss进行计算的时候碰到如标题所示的bug 原因分析:原因是没有实例化网络,直接调用类来进行forwardimport torchimport torch.nn as nnimport torch.nn.functional as Fimport mathfrom torchvision.models.vgg import vgg16import numpy as np'''Zero-DCE Spatial C
原创
2021-05-12 18:13:18 ·
2205 阅读 ·
1 评论