Continuing the discussion from Multi-class classification 5:



Multi-class classification


test_loader_X = torch.utils.data.DataLoader(dataset=tr_latent_X,
                                             batch_size=bs,
                                             shuffle=False)



It seems to me there is a bug in the code this should have said
test_loader_X = torch.utils.data.DataLoader(dataset=te_latent_X,
                                             batch_size=bs,
                                             shuffle=False)