model = ConvNet() #创建了一个实例
optimizer = optim.Adam(model.parameters(), lr=learning_rate) #优化函数,model.parameters()为该实例中可优化的参数,lr为参数优化的选项(学习率等) for name, param in model.named_parameters(): #查看可优化的参数有哪些 if param.requires_grad: print(name)
model = ConvNet() #创建了一个实例
optimizer = optim.Adam(model.parameters(), lr=learning_rate) #优化函数,model.parameters()为该实例中可优化的参数,lr为参数优化的选项(学习率等) for name, param in model.named_parameters(): #查看可优化的参数有哪些 if param.requires_grad: print(name)