🐛 fixing wrong type for growth_factor

rbturnbull · Sep 17, 2022 · 222225d · 222225d
1 parent c26407c
commit 222225d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/supercat/apps.py b/supercat/apps.py
@@ -219,12 +219,12 @@ def model(
             tune_max=256,
             help="The number of features after the initial CNN layer."
         ),
-        growth_factor:int = ta.Param(
+        growth_factor:float = ta.Param(
             2.0,
             tune=True, 
             tune_min=1.0,
             tune_max=4.0,
-            log=True,
+            #log=True,
             help="The factor to grow the number of convolutional filters each time the model downscales."
         ),
     ):