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