fixing typo. add option for 'momentum'.

nusdbsystem · Mar 14, 2021 · 22d08bc · 22d08bc
1 parent 141c95e
commit 22d08bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/models/image_classification/PyPandaResNet.py b/examples/models/image_classification/PyPandaResNet.py
@@ -64,10 +64,11 @@ def _create_model(self, scratch: bool, num_classes: int):
     @staticmethod
     def get_knob_config():
         return {
-        	'model_class':CategoricalKnob(['resnent101_mnist']),
+        	'model_class':CategoricalKnob(['resnet101_mnist']),
             # Learning parameters
             'lr':FixedKnob(0.0001), ### learning_rate
             'weight_decay':FixedKnob(0.0),
+            'momentum':FixedKnob(0),
             'drop_rate':FixedKnob(0.0),
             'max_epochs': FixedKnob(30), 
             'batch_size': CategoricalKnob([200]),