import torch
import numpy as np

from model import Net

a = np.arange(100).reshape(1, 100)
model = Net(100, 50, 10)
model.load_state_dict(torch.load('model.pth'))
model.eval()
output = model(torch.from_numpy(a).float())
print(output)