optimizer = optim.Adam(net.parameters(), lr=0.02,seed=0) TypeError: __init__() got an unexpected keyword argument 'seed'
时间: 2023-12-06 08:05:53 浏览: 130
TypeError: Unexpected keyword argument passed to optimizer: learning_rate 解决方法
这个错误是因为`optim.Adam`类不支持`seed`参数。在PyTorch中,带有`seed`参数的随机数生成需要使用`torch.manual_seed(seed)`来设定随机数种子。
如果你想要设置优化器的随机种子,可以在初始化随机数生成器后,在创建`optim.Adam`实例之前调用`torch.manual_seed(seed)`,例如:
```
import torch
import torch.optim as optim
torch.manual_seed(0)
net = MyModel()
optimizer = optim.Adam(net.parameters(), lr=0.02)
```
这样会将整个程序中的随机数生成都设定为相同的种子,从而保证每次运行程序得到的结果都一致。
阅读全文