94472724eedd3382396d435fe2810422f472967e,tests/squeezenet.py,,,#,11

Before Change

import torch.nn as nn
from torch.autograd import Variable

sys.path.append("../pytorch2keras")
from converter import pytorch_to_keras

// The code from torchvision
import math

After Change

        k_model = pytorch_to_keras(model, input_var, (3, 224, 224,), verbose=True)

        pytorch_output = output.data.numpy()
        keras_output = k_model.predict(input_np)

        error = np.max(pytorch_output - keras_output)
        print(error)
        if max_error < error:
            max_error = error

    print("Max error: {0}".format(max_error))

Instances