-
Notifications
You must be signed in to change notification settings - Fork 8
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
separate Sup3rGen and Sup3rGan #116
Conversation
sup3r/models/generator.py
Outdated
loss = loss_gen | ||
elif train_disc: | ||
loss = loss_disc | ||
# print("min max out = %g / %g " % |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
forgot to remove this
@@ -22,12 +21,11 @@ | |||
logger = logging.getLogger(__name__) | |||
|
|||
|
|||
class Sup3rGan(AbstractSup3rGan): | |||
"""Basic sup3r GAN model.""" | |||
class Sup3rGen(AbstractSup3rGan): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Separating this is a good idea
logger = logging.getLogger(__name__) | ||
|
||
|
||
class Sup3rGan(Sup3rGen): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Im wondering if we should still call this file base.py, since it includes the "base" Gan model. This would avoid all the renamed imports, although that alone shouldn't prevent an improvement.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, now that I think more about it, aside from the conditional moments models, all the models are GANs, so we can probably keep it as base
separate Sup3rGen and Sup3rGan
@malihass a few comments:
|
My plan was to make |
Gotcha yeah agree with everything you just said. Here's one structure that I think might be most clean: AbstractInterface (rename current AbstractSup3rGan) -> |
In prevision of the addition of the conditional moment models, make
Sup3rGan
inherit fromSup3rGen
which is a model that contains only the generator details.This way, the conditional moment model will be able to inherit from the
Sup3rGen
class and will not carry the extra attributes associated with the discriminator.I am opening the PR now in order to avoid large branch divergence since this PR changes how we call the
base
model