'SGD' object has no attribute 'build'
时间: 2023-09-03 08:10:26 浏览: 174
SGD.zip_SGD
这个错误是因为在您的代码中,'SGD'对象没有'build'属性。根据引用和引用的内容,这个错误通常是由于PyTorch版本更新引起的。在旧版本的PyTorch中,'SGD'对象有一个'defaults'属性,但在新版本中已经不再使用了。解决这个问题的方法是,确保您正在使用与训练模型时相同版本的PyTorch,并且保存优化器的参数。根据引用中的建议,您可以通过使用`torch.save(optimizer.state_dict(), PATH)`来保存优化器的状态字典。这样,在加载模型时,您可以使用`optimizer.load_state_dict(torch.load(PATH))`来加载优化器的状态。这样,您应该能够避免出现这个错误并成功保存和加载您的训练模型。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [pytorch保存模型的坑](https://blog.csdn.net/chenf1999/article/details/104333709)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文