Skip to content

Commit e3ecfed

Browse files
yeqinglitensorflower-gardener
authored andcommitted
Rename binary and updates README.
PiperOrigin-RevId: 365582100
1 parent d4b656b commit e3ecfed

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

official/vision/beta/projects/yt8m/README.md

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
# YouTube-8M Tensorflow Starter Code
22

3+
**WARNING**: The project is under construction. You can use it, but you do so
4+
at your own risk.
5+
36
This repo contains starter code (written in TensorFlow 2.x) for training and
47
evaluating machine learning models over the [YouTube-8M][1] dataset.
58
This is the Tensorflow2 version of the original starter code:
@@ -84,7 +87,7 @@ task:
8487
```
8588

8689
The code can be run in different modes: `train / train_and_eval / eval`.
87-
Run `yt8m_train.py` and specify which mode you wish to execute.
90+
Run `train.py` and specify which mode you wish to execute.
8891
Training is done using frame-level features with video-level labels,
8992
while inference can be done at segment-level.
9093
Setting `segment_labels=True` in your configuration forces
@@ -95,7 +98,7 @@ The following commands will train a model on Google Cloud over frame-level
9598
features.
9699

97100
```bash
98-
python3 yt8m_train.py --mode='train' \
101+
python3 train.py --mode='train' \
99102
--experiment='yt8m_experiment' \
100103
--model_dir=$MODEL_DIR \
101104
--config_file=$CONFIG_FILE
@@ -109,7 +112,7 @@ validation:`input_path=gs://youtube8m-ml/3/frame/validate/validate*.tfrecord`
109112
as default.
110113

111114
```bash
112-
python3 yt8m_train.py --mode='train_and_eval' \
115+
python3 train.py --mode='train_and_eval' \
113116
--experiment='yt8m_experiment' \
114117
--model_dir=$MODEL_DIR \
115118
--config_file=$CONFIG_FILE \
@@ -118,7 +121,7 @@ python3 yt8m_train.py --mode='train_and_eval' \
118121
Running on evaluation mode loads saved checkpoint from specified path
119122
and runs inference task.
120123
```bash
121-
python3 yt8m_train.py --mode='eval' \
124+
python3 train.py --mode='eval' \
122125
--experiment='yt8m_experiment' \
123126
--model_dir=$MODEL_DIR \
124127
--config_file=$CONFIG_FILE
File renamed without changes.

0 commit comments

Comments
 (0)