Python ModelSerializer.convert_param_to_cpu示例

示例#1

文件： train_chain_asr_ttsmspk.py 项目： templeblock/latest

 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)

示例#2

文件： train_vc_cyclegan_static.py 项目： templeblock/latest

 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