Update trainer.py

tbd-ai · Apr 3, 2018 · 894cfd2 · 894cfd2
1 parent 41117e4
commit 894cfd2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ObjectDetection-Faster_RCNN/TensorFlow/source/object_detection/trainer.py b/ObjectDetection-Faster_RCNN/TensorFlow/source/object_detection/trainer.py
@@ -32,6 +32,7 @@
 from object_detection.utils import variables_helper
 from deployment import model_deploy
 from tensorflow.python.framework import constant_op
+from profile_step import train_step
 
 slim = tf.contrib.slim