Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "0a6f6426ae511c0a09ae5fe2c8cafdbecacdbd1b"
Commit ec6ac5fa authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Fix flag name error in NCF create_ncf_data binary. 

PiperOrigin-RevId: 264468498
parent a6f10ec0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/recommendation/create_ncf_data.py
View file @ ec6ac5fa
...@@ -65,8 +65,8 @@ def prepare_raw_data(flag_obj): ...@@ -65,8 +65,8 @@ def prepare_raw_data(flag_obj):
data_processing_params = { data_processing_params = {
"train_epochs": flag_obj.num_train_epochs, "train_epochs": flag_obj.num_train_epochs,
"batch_size": flag_obj.prebatch_size, "batch_size": flag_obj.train_prebatch_size,
"eval_batch_size": flag_obj.prebatch_size, "eval_batch_size": flag_obj.eval_prebatch_size,
"batches_per_step": 1, "batches_per_step": 1,
"stream_files": True, "stream_files": True,
"num_neg": flag_obj.num_negative_samples, "num_neg": flag_obj.num_negative_samples,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment