Fixup

undertherain · Jul 23, 2020 · 2a7ab49 · 2a7ab49
1 parent 675a0ad
commit 2a7ab49
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/benchmarker/modules/problems/cosmoflow/pytorch.py b/benchmarker/modules/problems/cosmoflow/pytorch.py
@@ -47,6 +47,7 @@ def build_model(input_shape, target_size, dropout=0):
         nn.Linear(128, 64),
         nn.LeakyReLU(),
         nn.Dropout(dropout),
+        #
         nn.Linear(64, target_size),
         nn.Tanh(),
         scale_1p2(),