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
d569b531
Commit
d569b531
authored
Dec 27, 2018
by
Taylor Robie
Browse files
remove references to hash pipeline
parent
4fb325da
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
6 deletions
+1
-6
official/recommendation/ncf_main.py
official/recommendation/ncf_main.py
+0
-1
official/recommendation/run.sh
official/recommendation/run.sh
+1
-5
No files found.
official/recommendation/ncf_main.py
View file @
d569b531
...
@@ -155,7 +155,6 @@ def parse_flags(flags_obj):
...
@@ -155,7 +155,6 @@ def parse_flags(flags_obj):
"train_epochs"
:
flags_obj
.
train_epochs
,
"train_epochs"
:
flags_obj
.
train_epochs
,
"batches_per_step"
:
num_devices
,
"batches_per_step"
:
num_devices
,
"use_seed"
:
flags_obj
.
seed
is
not
None
,
"use_seed"
:
flags_obj
.
seed
is
not
None
,
"hash_pipeline"
:
flags_obj
.
hash_pipeline
,
"batch_size"
:
batch_size
,
"batch_size"
:
batch_size
,
"eval_batch_size"
:
eval_batch_size
,
"eval_batch_size"
:
eval_batch_size
,
"learning_rate"
:
flags_obj
.
learning_rate
,
"learning_rate"
:
flags_obj
.
learning_rate
,
...
...
official/recommendation/run.sh
View file @
d569b531
...
@@ -54,11 +54,7 @@ do
...
@@ -54,11 +54,7 @@ do
# To reduce variation set the seed flag:
# To reduce variation set the seed flag:
# --seed ${i}
# --seed ${i}
#
# And to confirm that the pipeline is deterministic pass the flag:
# --hash_pipeline
#
# (`--hash_pipeline` will slow down training, though not as much as one might imagine.)
python ncf_main.py
--model_dir
${
MODEL_DIR
}
\
python ncf_main.py
--model_dir
${
MODEL_DIR
}
\
--data_dir
${
DATA_DIR
}
\
--data_dir
${
DATA_DIR
}
\
--dataset
${
DATASET
}
--hooks
""
\
--dataset
${
DATASET
}
--hooks
""
\
...
...
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