Как преобразовать модель и вес кафе в pytorch - PullRequest
Новая
       14

Как преобразовать модель и вес кафе в pytorch

0 голосов
/ 07 января 2020

Здравствуйте, мне нужна помощь в преобразовании веса и модели кафе в pytorch. Я попытался использовать github, который большинство других постов предлагают использовать github , но когда я использовал его, я столкнулся с большой проблемой, так как он использовал python 2, и в настоящее время я использую python 3. Я уже пытался удалить некоторый слой, который не покрывает github, вручную изменить старый синтаксис на новый синтаксис, но последняя ошибка ссылается на модуль nn из pytorch, и я понятия не имею, чтобы это исправить. 

Traceback (most recent call last):
  File "caffe2pytorch.py", line 30, in <module>
    pytorch_blobs, pytorch_models = forward_pytorch(protofile, weightfile)
  File "caffe2pytorch.py", line 17, in forward_pytorch
    net = caffenet.CaffeNet(protofile)
  File "/home/cgal/reference/SfSNet/caffe2pytorch/caffenet.py", line 384, in __init__
    self.add_module(name, model)
  File "/home/cgal/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 186, in add_module
    raise KeyError("module name can't contain \".\"")
KeyError: 'module name can\'t contain "."'

Так есть ли какие-либо предложения о том, как преобразовать вес и модель кофе в pytorch?

Это модель кафе, которую я хочу конвертировать скачать здесь

...