Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
9369b1a3
Commit
9369b1a3
authored
Mar 29, 2021
by
Yeqing Li
Committed by
A. Unique TensorFlower
Mar 29, 2021
Browse files
Rename binary and updates README.
PiperOrigin-RevId: 365582100
parent
fcbf2288
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
4 deletions
+7
-4
official/vision/beta/projects/yt8m/README.md
official/vision/beta/projects/yt8m/README.md
+7
-4
official/vision/beta/projects/yt8m/train.py
official/vision/beta/projects/yt8m/train.py
+0
-0
No files found.
official/vision/beta/projects/yt8m/README.md
View file @
9369b1a3
# YouTube-8M Tensorflow Starter Code
# YouTube-8M Tensorflow Starter Code
**WARNING**
: The project is under construction. You can use it, but you do so
at your own risk.
This repo contains starter code (written in TensorFlow 2.x) for training and
This repo contains starter code (written in TensorFlow 2.x) for training and
evaluating machine learning models over the
[
YouTube-8M
][
1
]
dataset.
evaluating machine learning models over the
[
YouTube-8M
][
1
]
dataset.
This is the Tensorflow2 version of the original starter code:
This is the Tensorflow2 version of the original starter code:
...
@@ -84,7 +87,7 @@ task:
...
@@ -84,7 +87,7 @@ task:
```
```
The code can be run in different modes:
`train / train_and_eval / eval`
.
The code can be run in different modes:
`train / train_and_eval / eval`
.
Run
`
yt8m_
train.py`
and specify which mode you wish to execute.
Run
`train.py`
and specify which mode you wish to execute.
Training is done using frame-level features with video-level labels,
Training is done using frame-level features with video-level labels,
while inference can be done at segment-level.
while inference can be done at segment-level.
Setting
`segment_labels=True`
in your configuration forces
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
...
@@ -95,7 +98,7 @@ The following commands will train a model on Google Cloud over frame-level
features.
features.
```
bash
```
bash
python3
yt8m_
train.py
--mode
=
'train'
\
python3 train.py
--mode
=
'train'
\
--experiment
=
'yt8m_experiment'
\
--experiment
=
'yt8m_experiment'
\
--model_dir
=
$MODEL_DIR
\
--model_dir
=
$MODEL_DIR
\
--config_file
=
$CONFIG_FILE
--config_file
=
$CONFIG_FILE
...
@@ -109,7 +112,7 @@ validation:`input_path=gs://youtube8m-ml/3/frame/validate/validate*.tfrecord`
...
@@ -109,7 +112,7 @@ validation:`input_path=gs://youtube8m-ml/3/frame/validate/validate*.tfrecord`
as default.
as default.
```
bash
```
bash
python3
yt8m_
train.py
--mode
=
'train_and_eval'
\
python3 train.py
--mode
=
'train_and_eval'
\
--experiment
=
'yt8m_experiment'
\
--experiment
=
'yt8m_experiment'
\
--model_dir
=
$MODEL_DIR
\
--model_dir
=
$MODEL_DIR
\
--config_file
=
$CONFIG_FILE
\
--config_file
=
$CONFIG_FILE
\
...
@@ -118,7 +121,7 @@ python3 yt8m_train.py --mode='train_and_eval' \
...
@@ -118,7 +121,7 @@ python3 yt8m_train.py --mode='train_and_eval' \
Running on evaluation mode loads saved checkpoint from specified path
Running on evaluation mode loads saved checkpoint from specified path
and runs inference task.
and runs inference task.
```
bash
```
bash
python3
yt8m_
train.py
--mode
=
'eval'
\
python3 train.py
--mode
=
'eval'
\
--experiment
=
'yt8m_experiment'
\
--experiment
=
'yt8m_experiment'
\
--model_dir
=
$MODEL_DIR
\
--model_dir
=
$MODEL_DIR
\
--config_file
=
$CONFIG_FILE
--config_file
=
$CONFIG_FILE
...
...
official/vision/beta/projects/yt8m/
yt8m_
train.py
→
official/vision/beta/projects/yt8m/train.py
View file @
9369b1a3
File moved
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment