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
GANModel Porting #3601
GANModel Porting #3601
Conversation
So presently i am using Tensorflow Keras tensors have a method The 2 tests checking for the mean of deltas are the ones failing |
deepchem/models/torch_models/gan.py
Outdated
|
||
This class allows a GAN to have multiple generators and discriminators, a model | ||
known as MIX+GAN. It is described in Arora et al., "Generalization and | ||
Equilibrium in Generative Adversarial Nets (GANs)" (https://arxiv.org/abs/1703.00573). |
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.
Link also to the reference below
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.
done
deepchem/models/torch_models/gan.py
Outdated
|
||
Examples | ||
-------- | ||
<examples here> |
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.
Add examples here before final review
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.
done
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.
@shreyasvinaya This looks pretty clean. Are we fully matching the existing TF API? How are the number?
If API is the same and numbers are comparable we can probably merge in tomorrow
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.
LGTM
Description
Porting of GANModel to Torch
Type of change
Please check the option that is related to your PR.
Checklist
yapf -i <modified file>
and check no errors (yapf version must be 0.32.0)mypy -p deepchem
and check no errorsflake8 <modified file> --count
and check no errorspython -m doctest <modified file>
and check no errors