swin_transformer_v2.py RuntimeError 期望所有张量都位于同一设备,但发现至少有两个设备,分别是 cuda:0 和 cpu! (在检查方法 wrapper_CUDA_clamp_Tensor 中的参数 max 的参数时)

使用 swin-Transformer-v2 训练时，出现了 RuntimeError: 期望所有张量都在同一设备上，但发现至少有两个设备，cuda:0 和 cpu！(在检查方法 wrapper_CUDA_clamp_Tensor 中参数 max 的参数时发生。) 以下是修复 RuntimeError 的 Pull Request 链接。