训练数据的时候出现

RuntimeError:CUDA error:device-side assert triggeredCUDA kernel errors
might be asynchronously reported at some other API call, so the
stacktrace below might be incorrect.For debugging consider passing
CUDA_LAUNCH_BLOCKING=1.Compile with TORCH USE CUDA DSA to enable
device-side assertions.

检查一下数据集标签和训练时的标签是否一致。