"vscode:/vscode.git/clone" did not exist on "c47418f0da3c756eb65ca7413af7d77b4ec6725f"
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
......@@ -65,8 +65,8 @@ def prepare_raw_data(flag_obj):
data_processing_params = {
"train_epochs": flag_obj.num_train_epochs,
"batch_size": flag_obj.prebatch_size,
"eval_batch_size": flag_obj.prebatch_size,
"batch_size": flag_obj.train_prebatch_size,
"eval_batch_size": flag_obj.eval_prebatch_size,
"batches_per_step": 1,
"stream_files": True,
"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