_copy_to_cpu not defined

YanjieZe / 3D-Diffusion-Policy

[RSS 2024] 3D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representations

https://3d-diffusion-policy.github.io

MIT License

402 stars 37 forks source link

Closed Kimagure7 closed 3 months ago

Kimagure7 commented 4 months ago

line 399 in train.py: payload['state_dicts'][key] = _copy_to_cpu(value.state_dict()) _copy_to_cpu not defined

YanjieZe commented 3 months ago

Hi, thank you for your interest. I have not encountered this problem before. Have you solved it yet?