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)