Loss.cu:97,RuntimeError: copy_if failed to synchronize: cudaErrorAssert: device-side assert triggere

NoSuchKey