I am running siamese network which uses contrastive loss function and SGD optimiser with lr = 0.1,momentum= 0.9,and weight_decay = 0.0005
unspecified launch failure at /opt/conda/conda-bld/pytorch_1544174967633/work/aten/src/THC/generated/…/THCReduceAll.cuh:317
My code was fine for 90 epochs and after which it generated the above error.