Unexpected keyword argument passed to optimizer: weight_decay
时间: 2024-01-12 13:11:43 浏览: 356
TypeError: Unexpected keyword argument passed to optimizer: learning_rate 解决方法
这个错误通常是因为你在使用的优化器中传递了一个不被支持的参数。在 PyTorch 中,有些优化器(如 AdamW)支持 weight_decay 参数,而有些优化器(如 Adam)不支持。因此,如果你在使用 Adam 优化器时尝试传递 weight_decay 参数,就会出现这个错误。
解决方法是检查你正在使用的优化器是否支持 weight_decay 参数,如果不支持,则需要使用另一种支持该参数的优化器。如果你确信你使用的优化器支持 weight_decay 参数,那么可以检查一下你的代码是否正确传递了该参数。
阅读全文