def save_model(model, model_path, model_name):
torch.save(ModelSerializer.convert_param_to_cpu(model.state_dict()),
os.path.join(model_path, '{}.mdl'.format(model_name)))
ModelSerializer.save_config(
os.path.join(model_path, '{}.cfg'.format(model_name)),
model.get_config())
json.dump(opts,
open(os.path.join(opts['result'], 'script.opts'), 'w'),
indent=2)
def save_model(model, name, epoch) :
torch.save(ModelSerializer.convert_param_to_cpu(model.state_dict()), os.path.join(opts['result'], 'model_{}_e{}.mdl'.format(name, epoch)))
ModelSerializer.save_config(os.path.join(opts['result'], 'model_{}_e{}.cfg'.format(name, epoch)), model.get_config())
pass