Open iwasnotyet opened 1 year ago
确实是个问题,原因有两个:
empty_cuda_cache
时应该先看看当前 device
是不是 cpu
TranslatorAPI
也需要实现一个 to
方法,从而在需要时可以把它放到 cuda
上执行我会尽快进行修复,修好后再让你试试😉
刚刚把这个问题修复了,不过需要重新 pull 一下代码以及重新 pip install -e .
一下(因为需要更新背后 carefree-learn
的版本);如果还有问题的话,欢迎继续留言😆
刚刚把这个问题修复了,不过需要重新 pull 一下代码以及重新
pip install -e .
一下(因为需要更新背后carefree-learn
的版本);如果还有问题的话,欢迎继续留言😆
十分感谢!