Skip to content

Commit

Permalink
Update dcgan.py
Browse files Browse the repository at this point in the history
Changed the dimensions of the hidden channels
  • Loading branch information
HarshithBachimanchi committed May 5, 2024
1 parent e553bed commit ef67856
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions deeplay/models/generators/dcgan.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,11 +86,11 @@ def __init__(
class_conditioned_model: bool = False,
embedding_dim: int = 100,
num_classes: int = 10,
output_channels=None
output_channels=None,
):
if output_channels is not None:
out_channels = output_channels

self.latent_dim = latent_dim
self.output_channels = out_channels
self.class_conditioned_model = class_conditioned_model
Expand All @@ -104,10 +104,10 @@ def __init__(
super().__init__(
in_channels=in_channels,
hidden_channels=[
features_dim * 16,
features_dim * 8,
features_dim * 4,
features_dim * 2,
features_dim * 1,
],
out_channels=out_channels,
out_activation=Layer(nn.Tanh),
Expand Down

0 comments on commit ef67856

Please sign in to comment.