在训练模型期间,我将一些训练参数保存在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
中的变量?