search

moskomule / anatome

Ἀνατομή is a PyTorch library to analyze representation of neural networks
MIT License
61 stars 6 forks source link

Use FX instead of hooks #25

Closed moskomule closed 2 years ago

moskomule commented 2 years ago

torch.fx is now stable.

brando90 commented 2 years ago

torch.fx is now stable.

curious, how did you find out about torch.fx? (I'm also curious if there are any advantages to using that vs hooks)