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)

op_insertion.py

@@ -204,7 +204,6 @@ def fq_creation(input_tensor, name, init_value=6):
             tf.io.write_graph(concrete.graph, '/tmp', 'mobilenetv2_sub_with_conv.pb')
 
     def call(self, inputs, training=None):
-        training = True
         model_obj = self.trainable_model if training else self.eval_model
         if isinstance(tf.distribute.get_strategy(), tf.distribute.MirroredStrategy):
             replica_context = tf.distribute.get_replica_context()