Как я могу исправить размер Param? - PullRequest
0 голосов
/ 21 сентября 2019

Я хочу сделать веб-сервис, который использует COCO API.Поэтому я делаю веб-сервер с помощью вычислительной системы Google Cloud Platform.Я использовал Linux Debian9 Python3.после исправления я запускаю 3_Inference.py (python3 Inference.py).и встретите это сообщение об ошибке.

"
Vocabulary successfully loaded from vocab.pkl file!
Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
Traceback (most recent call last):
  File "20190912_test.py", line 77, in <module>
    decoder.load_state_dict(checkpoint['decoder'])
  File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 845, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DecoderRNN:
        size mismatch for embed.weight: copying a param with shape torch.Size([6335, 256]) from checkpoint, the shape in current model is torch.Size([8855, 256]).
        size mismatch for linear.bias: copying a param with shape torch.Size([6335]) from checkpoint, the shape in current model is torch.Size([8855]).
        size mismatch for linear.weight: copying a param with shape torch.Size([6335, 512]) from checkpoint, the shape in current model is torch.Size([8855, 512]).
"

На самом деле я создал файл model.py на своем локальном компьютере, а теперь в VM (GCP Compute Engine) я использовал этот model.py.

Мне снова нужно пройти этап обучения и проверки?

Нет ли способа исправить размер параметра?

Пожалуйста, дайте мне знать.Спасибо.

...