Setup training

daniil-lyakhov · Jul 22, 2021 · c327d1b · c327d1b
1 parent 027e7f3
commit c327d1b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/examples/tensorflow/classification/configs/quantization/mobilenet_v2_imagenet_int8.json b/examples/tensorflow/classification/configs/quantization/mobilenet_v2_imagenet_int8.json
@@ -6,16 +6,16 @@
     },
 
     "batch_size": 256,
-    "epochs": 18,
+    "epochs": 16,
     "num_classes": 1001,
     "dataset_preprocessing_preset": "imagenet2012_slim",
 
     "optimizer": {
         "type": "Adam",
         "schedule_type": "piecewise_constant",
         "schedule_params": {
-            "boundaries": [8, 12, 15],
-            "values": [5e-4, 5e-5, 5e-6, 1e-6]
+            "boundaries": [6, 10],
+            "values": [1e-4, 1e-5, 1e-6]
         }
     },
 

diff --git a/examples/tensorflow/classification/main.py b/examples/tensorflow/classification/main.py
@@ -306,8 +306,8 @@ def run(config):
         logger.info('training...')
         compress_model.fit(
             train_dataset,
-            epochs=1,#train_epochs,
-            steps_per_epoch=3,#train_steps,
+            epochs=train_epochs,
+            steps_per_epoch=train_steps,
             initial_epoch=initial_epoch,
             callbacks=callbacks,
             **validation_kwargs)