pytorch-vgg icon indicating copy to clipboard operation
pytorch-vgg copied to clipboard

AssertionError: y_diff = 0.000188 when converting vgg16 and vgg19 models

Open ProGamerGov opened this issue 7 years ago • 0 comments

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

ProGamerGov avatar May 04 '18 00:05 ProGamerGov