python-3.x - 如何加载保存的 argparser 并覆盖当前的？

Question

在训练我的模型期间，我在 argparser 中保存了一些训练参数，例如：

torch.save({'iteration': iter,
            'model': model.state_dict(),
            'voc': voc,
            'args': args
            }, 'path')))

我想知道如何加载保存args并（更重要的是）覆盖当前args？例如，</p>

from conf import args
train_args = torch.load(args.path)['args']
args = train_args

但是，上面的代码是无效的。如何用 in args中的相应变量覆盖 in 中的变量train_args？

标签： python-3.xpytorch