• torch.load( )
  • model.load_state_dict( ) ```python import torch

定义模型

rnn = GRU_plain(input_size=1,embedding_size = 64).cuda

加载模型参数的文件

rnn.load_state_dict(torch.load(‘rnn_1.dat’)) ```