Loss statistics -
({'train': [tensor(0.4949, device='cuda:0', grad_fn=
),
tensor(0.0581, device='cuda:0', grad_fn=),
tensor(0.0483, device='cuda:0', grad_fn=),
tensor(0.0270, device='cuda:0', grad_fn=)],
'val': [tensor(0.0624, device='cuda:0'),
tensor(0.0402, device='cuda:0'),
tensor(0.0741, device='cuda:0'),
tensor(0.0964, device='cuda:0')]},Accuracy statistics -
{'train': [tensor(0.8138, device='cuda:0'),
tensor(0.9858, device='cuda:0'),
tensor(0.9890, device='cuda:0'),
tensor(0.9916, device='cuda:0')],
'val': [tensor(0.9792, device='cuda:0'),
tensor(0.9848, device='cuda:0'),
tensor(0.9867, device='cuda:0'),
tensor(0.9811, device='cuda:0')]})
