我希望我的培训计划能够实现“培训”的功能。但我不知道如何正确恢复优化器状态。
我的程序是这样的:
opt = mx.optimizer(learning_rate = lr,…
我会尝试使用 load_checkpoint 这样做的功能。看看在这个例子中它是如何完成的: https://mxnet.incubator.apache.org/versions/master/tutorials/basic/module.html#save-and-load
load_checkpoint