Setup training

Author: daniil-lyakhov

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]
         }
     },
 

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)