def infer(args):
data = Data(str(args.test))
model = Project(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()
data_loader = torch.utils.data.DataLoader(
data, batch_size=len(data), num_workers=24
)
for i in range(data_loader):
results = model(data_loader)
file = open(args.testcsv, "w")
file.write("\n".join(results.tolist()))
file.close()
print("Done....")
