class Easymodel(nn.Module):
def __init__(self):
self.name = 1
def forward(self,x ):
return x + 1
model = Easymodel()
x = torch.rand(3,3,8,8).to(device)
out = model(x)
原因是这里面没有任何一个需要梯度的地方
模型没有参数，数据也不需要求梯度
class Easymodel(nn.Module):
def __init__(self):
self.name = 1
def forward(self,x ):
return x + 1
model = Easymodel()
x = torch.rand(3,3,8,8).to(device)
out = model(x)
原因是这里面没有任何一个需要梯度的地方
模型没有参数，数据也不需要求梯度
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
