是pytorch1.12的bug,官方在1.12.1进行了修复,可以使用1.11.
If capturable=False, state_steps should not be CUDA tensors.
最新推荐文章于 2024-07-06 22:04:51 发布
是pytorch1.12的bug,官方在1.12.1进行了修复,可以使用1.11.