Retrieve deeply nested hooks by name

KevinMusgrave / pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.

https://kevinmusgrave.github.io/pytorch-adapt/

MIT License

359 stars 15 forks source link

Open KevinMusgrave opened 2 years ago

KevinMusgrave commented 2 years ago

For example:

# return all hooks inside of "classifier" that are of type ATDOCHook
hooks = classifier.get_hooks("ATDOCHook")