-
-
Notifications
You must be signed in to change notification settings - Fork 46.7k
Generative Adversarial Network for MNIST Dataset #11961
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
Conversation
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.
Click here to look at the relevant links ⬇️
🔗 Relevant Links
Repository:
Python:
Automated review generated by algorithms-keeper. If there's any problem regarding this review, please open an issue about it.
algorithms-keeper
commands and options
algorithms-keeper actions can be triggered by commenting on this PR:
@algorithms-keeper review
to trigger the checks for only added pull request files@algorithms-keeper review-all
to trigger the checks for all the pull request files, including the modified files. As we cannot post review comments on lines not part of the diff, this command will post all the messages in one comment.NOTE: Commands are in beta and so this feature is restricted only to a member or owner of the organization.
|
||
# Creating Generator and Discriminator for GAN | ||
|
||
class discriminator(nn.Module): |
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.
Class names should follow the CamelCase
naming convention. Please update the following name accordingly: discriminator
# Creating Generator and Discriminator for GAN | ||
|
||
class discriminator(nn.Module): | ||
def __init__(self,input_size,output_size,hidden_dim): |
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.
Please provide return type hint for the function: __init__
. If the function does not return a value, please provide the type hint as: def function() -> None:
Please provide type hint for the parameter: input_size
Please provide type hint for the parameter: output_size
Please provide type hint for the parameter: hidden_dim
#dropout layer | ||
self.dropout = nn.Dropout(0.2) | ||
|
||
def forward(self,x): |
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.
Please provide return type hint for the function: forward
. If the function does not return a value, please provide the type hint as: def function() -> None:
As there is no test file in this pull request nor any test function or class in the file computer_vision/Generative_Adversarial_Network_MNIST.py
, please provide doctest for the function forward
Please provide type hint for the parameter: x
Please provide descriptive name for the parameter: x
|
||
return x_out | ||
|
||
class generator(nn.Module): |
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.
Class names should follow the CamelCase
naming convention. Please update the following name accordingly: generator
|
||
class generator(nn.Module): | ||
|
||
def __init__(self, input_size, output_size,hidden_dim): |
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.
Please provide return type hint for the function: __init__
. If the function does not return a value, please provide the type hint as: def function() -> None:
Please provide type hint for the parameter: input_size
Please provide type hint for the parameter: output_size
Please provide type hint for the parameter: hidden_dim
|
||
# Compute the discriminator losses on real images | ||
# use smoothed labels | ||
D_real = D(real_images) |
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.
Variable and function names should follow the snake_case
naming convention. Please update the following name accordingly: D_real
fake_images = G(z) | ||
|
||
# Compute the discriminator losses on fake images | ||
D_fake = D(fake_images) |
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.
Variable and function names should follow the snake_case
naming convention. Please update the following name accordingly: D_fake
fake_images = G(z) | ||
# Compute the discriminator losses on fake images | ||
# using flipped labels! | ||
D_fake = D(fake_images) |
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.
Variable and function names should follow the snake_case
naming convention. Please update the following name accordingly: D_fake
# Print some loss stats | ||
if batch_i % print_every == 0: | ||
# print discriminator and generator loss | ||
print('Epoch [{:5d}/{:5d}] | d_loss: {:6.4f} | g_loss: {:6.4f}'.format( |
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.
As mentioned in the Contributing Guidelines, please do not use printf style formatting or str.format()
. Use f-string instead to be more readable and efficient.
|
||
|
||
#Viewing the results of the GAN | ||
def view_samples(epoch, samples): |
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.
Please provide return type hint for the function: view_samples
. If the function does not return a value, please provide the type hint as: def function() -> None:
As there is no test file in this pull request nor any test function or class in the file computer_vision/Generative_Adversarial_Network_MNIST.py
, please provide doctest for the function view_samples
Please provide type hint for the parameter: epoch
Please provide type hint for the parameter: samples
for more information, see https://pre-commit.ci
Closing require_type_hints PRs to prepare for Hacktoberfest |
Describe your change:
Checklist: