我试图转换CNN Keras模型使用FER2013数据集PyTorch模型情感识别和我有以下错误:
Traceback (most recent call last):
File "VGG.py", line 112, in <module>
transfer.keras_to_pytorch(keras_network, pytorch_network)
File "/home/eorg/NeuralNetworks/user/Project/model/nntransfer.py", line 121, in keras_to_pytorch
pytorch_model.load_state_dict(state_dict)
File "/home/eorg/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 334, in load_state_dict
own_state[name].copy_(param)
RuntimeError: inconsistent tensor size at /b/wheel/pytorch-src/torch/lib/TH/generic/THTensorCopy.c:51
Run Code Online (Sandbox Code Playgroud)
我知道错误与图像的形状有关。在 Keras 中,输入大小定义为 48 x 48。
我的问题是如何在 PyTorch 模型中定义我的图片是 48x48 的形状?我在文档和示例中找不到这样的功能。
任何帮助都会有用!