I saw:
https://pytorch.org/docs/stable/fx.html
I am using hooks to get features for CCA comparison on NNs (GitHub - brando90/ultimate-anatome: Ἀνατομή is a PyTorch library to analyze representation of neural networks). Is there an advantage of moving to migrating to torch.fx? if yes what are they?
Thanks in advance!

context:


Use FX instead of hooks · Issue #25 · moskomule/anatome · GitHub 1 and

GitHub - brando90/ultimate-anatome: Ἀνατομή is a PyTorch library to analyze representation of neural networks  and
torch.fx · Issue #9 · brando90/ultimate-anatome · GitHub