Does monai.engines.AdversarialTrainer Support torch.compile()? #8595
Tyler-Slater
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Apologies if this has been addressed before. I noticed that the SupervisedTrainer engine accepts a kwarg called "compile", but this kwarg is absent from AdversarialTrainer.
If I want to implement this manually by instantiating my generator and discriminator and calling torch.compile() on them before passing them into AdversarialTrainer, is this likely to cause issues?
Beta Was this translation helpful? Give feedback.
All reactions