0

我创建了一个 ModelCheckpoint() 实例,指定检查点应保存在何处以及在何种条件下,并将其用作 Trainer() 工具中的“回调”参数,但在完成一个训练时期后(我可以看到完成并返回了一个 val_loss)它无法保存在我的位置,因为:

AttributeError: 'Version' object has no attribute 'release'

它正在显示atomic_save代码并尝试评估以下内容:

if Version(torch.__version__).release[:3] == (1, 6, 0):

我已经看了一遍,这个特定的 AttributeError 似乎很不常见。我已经确保我的所有软件包都已更新。任何帮助表示赞赏。不确定我是否在这里遗漏了一些明显的东西。

4

0 回答 0