xinet training running

micromind-toolkit · Nov 27, 2023 · e62e761 · e62e761
1 parent 7f0f0bd
commit e62e761
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/recipes/image_classification/cfg/phinet.py b/recipes/image_classification/cfg/phinet.py
@@ -6,6 +6,7 @@
 """
 
 # Model configuration
+model = "phinet"
 input_shape = (3, 32, 32)
 alpha = 3
 num_layers = 7

diff --git a/recipes/image_classification/cfg/xinet.py b/recipes/image_classification/cfg/xinet.py
@@ -6,11 +6,12 @@
 """
 
 # Model configuration
-input_shape = (3, 32, 32)
+model = "xinet"
+input_shape = (3, 128, 128)
 alpha = 1
 num_layers = 7
 return_layers = None
-compression = 4
+gamma = 4
 
 ckpt_pretrained = ""
 

diff --git a/recipes/image_classification/train.py b/recipes/image_classification/train.py
@@ -55,7 +55,7 @@ def __init__(self, hparams, *args, **kwargs):
             self.modules["classifier"] = XiNet(
                 input_shape=hparams.input_shape,
                 alpha=hparams.alpha,
-                compression=hparams.compression,
+                gamma=hparams.gamma,
                 num_layers=hparams.num_layers,
                 return_layers=hparams.return_layers,
                 # classification-specific