You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/source/training.rst
+3-3
Original file line number
Diff line number
Diff line change
@@ -612,7 +612,7 @@ Once the above changes have been applied to our config file, go ahead and save i
612
612
Training the Model
613
613
~~~~~~~~~~~~~~~~~~
614
614
615
-
Before we begin training our model, let's go and copy the ``TensorFlow/models/research/object_detection/train.py`` script and paste it straight into our ``training_demo`` folder. We will need this script in order to train our model.
615
+
Before we begin training our model, let's go and copy the ``TensorFlow/models/research/object_detection/legacy/train.py`` script and paste it straight into our ``training_demo`` folder. We will need this script in order to train our model.
616
616
617
617
Now, to initiate a new training job, ``cd`` inside the ``training_demo`` folder and type the following:
618
618
@@ -715,10 +715,10 @@ Once your training job is complete, you need to extract the newly trained infere
715
715
716
716
activate tensorflow_gpu
717
717
718
-
- Copy the ``TensorFlow/models/research/object_detection/extract_inference_graph.py`` script and paste it straight into your ``training_demo`` folder.
718
+
- Copy the ``TensorFlow/models/research/object_detection/export_inference_graph.py`` script and paste it straight into your ``training_demo`` folder.
719
719
- Check inside your ``training_demo/training`` folder for the ``model.ckpt-*`` checkpoint file with the highest number following the name of the dash e.g. ``model.ckpt-34350``). This number represents the training step index at which the file was created.
720
720
- Alternatively, simply sort all the files inside ``training_demo/training`` by descending time and pick the ``model.ckpt-*`` file that comes first in the list.
721
-
- Make a note of the file's name, as it will be passed as an argument when we call the ``extract_inference_graph.py`` script.
721
+
- Make a note of the file's name, as it will be passed as an argument when we call the ``export_inference_graph.py`` script.
722
722
- Now, ``cd`` inside your ``training_demo`` folder, and run the following command:
0 commit comments