Создание модели Pytorch из существующего файла .PTH - PullRequest
0 голосов
/ 21 июня 2020

В результате сеанса обучения GAN я создал генератор midi и дискриминатор midi в виде файлов .PTH и хочу использовать генератор для создания midi с использованием модели pytorch.

При поиске в документации 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. Что мне нужно сделать, чтобы сгенерировать этот миди с помощью этой модели?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...