inference.py

import torch


from utils import show_predictions
from dataset import test_dataloader



# show_predictions(test_dataloader, model)

# x, y = next(iter(test_dataloader))




# print(x.shape)
# print(model)