Is there any method to save computation graph in pytorch? Just like tensorflow could save graph…
The typical thing is to use the JIT to trace or script your model. This is - at a very high level - somewhat similar as the computational graph in TensorFlow.
Thanks for reply, I think it’s really helpful for me!