siamese_train = SiameseDataset(mnist_train, transform) siamese_test = SiameseDataset(mnist_test, transform)
Define train and test dataset