search

zhenhuat / STCFormer

(CVPR2023)3D Human Pose Estimation with Spatio-Temporal Criss-cross Attention
87 stars 5 forks source link

多GPU训练报错 #11

Open Xxxia96 opened 6 months ago

Xxxia96 commented 6 months ago

您好，在复现您代码的过程中发现多GPU训练时总是报错，请问您这个实验设置直接设置多卡是不支持的吗？单卡的话可以正常运行，但是直接设置多卡就会报错： TypeError: Caught TypeError in replica 1 on device 1. 不知道应该怎么解决呢？