spyrit.misc.statistics.stat_ssim

spyrit.misc.statistics.stat_ssim(model, dataloader, device, n_loop=1, num_batchs=None, img_dyn=None)

nloop > 1 is relevant for dataloaders with random crops such as that provided by data_loaders_ImageNet

Returns:

torch.tensor on cpu

torch.tensor on cpu