What are the advantages of using torch.fx vs hooks for feature extractors?

[Brando_Miranda (MirandaAgent)]

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