pytorch-vgg
pytorch-vgg copied to clipboard
AssertionError: y_diff = 0.000188 when converting vgg16 and vgg19 models
th caffemodel_to_t7.lua -output_t7 VGG_ILSVRC_19_layers.t7 -input_caffemodel VGG_ILSVRC_19_layers.caffemodel -input_prototxt VGG_ILSVRC_19_layers_deploy.prototxt
python t7_to_state_dict.py --input_t7 VGG_ILSVRC_19_layers.t7 --model_name vgg19
python test_model.py --t7_file VGG_ILSVRC_19_layers.t7 --pth_file vgg19-bdbb0f7f.pth
Upon running the test_model.py script, I get this error:
Running test case 1 / 10
Traceback (most recent call last):
File "test_model.py", line 41, in <module>
assert y_diff == 0, 'y_diff = %f' % y_diff
AssertionError: y_diff = 0.000188
For the vgg16 model:
Running test case 1 / 10
Traceback (most recent call last):
File "test_model.py", line 41, in <module>
assert y_diff == 0, 'y_diff = %f' % y_diff
AssertionError: y_diff = 0.000198