如何加载已保存的argparser并覆盖当前的argparser?

时间:2019-06-26 09:00:41

标签: python-3.x pytorch

在训练模型期间,我将一些训练参数保存在argparser中,例如:

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

我想知道如何加载保存的args并(更重要的是)覆盖当前的args?例如,

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

但是,以上代码无效。如何用args中的相应变量覆盖train_args中的变量?

0 个答案:

没有答案