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

        input_np = np.random.uniform(0, 1, (1, 3, 224, 224))
        input_var = Variable(torch.FloatTensor(input_np))
        output = model(input_var)

        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)

Instances