import torch

from model import *

from  Auto_Json import run
model=S1()
# torch.save(model.state_dict(),"./ftrt/best.pth")
model.load_state_dict(torch.load("./ftrt/best.pth"))
model.eval()
a=torch.ones([1,3,32,32])
print(model(a))

run(model.to("cuda"),a.to("cuda"))