Question

作为GAN培训的结果，我生成了一个Midi生成器和一个Midi鉴别器作为.PTH文件，并希望使用该生成器通过pytorch模型创建一个Midi。

在查看PyTorch文档时，我发现了这段代码来描述如何从该.PTH文件加载模型。这就是我发现的。

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()

我使用以下代码重新创建了此代码：

import torch 

model = TheModelClass(*args,**kwargs)
model.load_state_dict(torch.load("C:\\Users\\ryand\\OneDrive\\Desktop\\c_rnn_gan_g.pth"))
model.eval()

这产生了以下错误：

---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
<ipython-input-12-8a77f4d4790c> in <module>
----> 1 import torch
      2 
      3 model = TheModelClass(*args,**kwargs)
      4 model.load_state_dict(torch.load("C:\\Users\\ryand\\OneDrive\\Desktop\\c_rnn_gan_g.pth"))
      5 model.eval()

~\Anaconda3\lib\site-packages\torch\__init__.py in <module>
     79     dlls = glob.glob(os.path.join(th_dll_path, '*.dll'))
     80     for dll in dlls:
---> 81         ctypes.CDLL(dll)
     82 
     83 

~\Anaconda3\lib\ctypes\__init__.py in __init__(self, name, mode, handle, use_errno, use_last_error)
    362 
    363         if handle is None:
--> 364             self._handle = _dlopen(self._name, mode)
    365         else:
    366             self._handle = handle

OSError: [WinError 126] The specified module could not be found

这是在安装了PyTorch的anaconda环境中运行的。我该怎么做才能用该模型生成Midi？

从现有的.PTH文件创建Pytorch模型

0 个答案: