train_dataloader = DataLoader(siamese_train, shuffle=True, num_workers=8, batch_size=64)
model = SiameseNetwork().cuda()
criterion = ContrastiveLoss()
optimizer = optim.Adam(model.parameters(), lr = 0.001)