Similar Problem, always around 7 epoch then throw a error
RuntimeError: reduce failed to synchronize: cudaErrorAssert: device-side assert triggered
the loss function is BCE , the optimize is Adam.
Have you solve the problem?