hzwer / Practical-RIFE

We are developing more practical frame interpolation approach.
MIT License
547 stars 60 forks source link

如何转换模型为TorchScript格式? #20

Open pango99 opened 1 year ago

pango99 commented 1 year ago

你好:

我写了个python脚本来转换模型为torchscript格式(cvtTorchScript.zip),当我用torch.script.trace()函数转换时发生错误,报告系统内存不够(错误信息,torch.jit.trace.zip),然后我改用torch.jit.script()来转换,同样发生错误(错误信息torch.jit.script.zip),所以请问下我该如何转换模型为TorchScript格式,谢谢!