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

作为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?

iCMS 回答:从现有的.PTH文件创建Pytorch模型

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/2089473.html

大家都在问