How to save computation graph in pytorch?

1

Is there any method to save computation graph in pytorch? Just like tensorflow could save graph…

2

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.

1 Like
3

Thanks for reply, I think it’s really helpful for me!