问题描述
我有一个模型，该模型具有不同的循环部分，而其他部分则严格非循环部分。对于我的训练过程，我需要高阶导数。不幸的是，就目前而言（2020年9月4日），PyTorch does not implement在CUDA上为RNN提供了双重支持。我建议避免使用Jitted版本的RNNS。
有没有办法在cpu上运行模型的一部分，在gpu上运行模型的一部分？还是委托RNN向后传递给另一个设备？我知道，从理论上讲，我需要来回移动整个模型和至少一部分图形。
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）