spyrit.core.train.visualize_model

spyrit.core.train.visualize_model(model, dataloaders, device, suptitle='', colormap=<matplotlib.colors.LinearSegmentedColormap object>)[source]: Takes 8 images from the dataloader and shows side by side the input image and the reconstructed image