a bit better

Lightning-AI · Aug 1, 2024 · c39a740 · c39a740
1 parent ce02a32
commit c39a740
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightning_examples/basic-gan/gan.py b/lightning_examples/basic-gan/gan.py
@@ -88,7 +88,7 @@ def block(in_feat, out_feat, normalize=True):
             layers = [nn.Linear(in_feat, out_feat)]
             if normalize:
                 layers.append(nn.BatchNorm1d(out_feat, 0.8))
-            layers.append(nn.LeakyReLU(0.2, inplace=True))
+            layers.append(nn.LeakyReLU(0.01, inplace=True))
             return layers
 
         self.model = nn.Sequential(
@@ -259,7 +259,7 @@ def on_validation_epoch_end(self):
 trainer = pl.Trainer(
     accelerator="auto",
     devices=1,
-    max_epochs=15,
+    max_epochs=5,
 )
 trainer.fit(model, dm)