If I Trace a PyTorch Network on Cuda, can I use it on CPU?

前端 未结 0 1587
南旧
南旧 2020-12-31 20:02

I traced my Neural Network using torch.jit.trace on a CUDA-compatible GPU server. When I reloaded that Trace on the same server, I could reload it and use it fi

相关标签:
回答
  • 消灭零回复
提交回复
热议问题