Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "a2e7387eba111aa647537505e37a39146a65c431"
Commit bce4604a authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Merge pull request #8510 from nuka137:add-ncf-instruction 

PiperOrigin-RevId: 311746915
parents fe4e6b65 7454645b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/recommendation/README.md
View file @ bce4604a
...@@ -67,5 +67,6 @@ Arguments: ...@@ -67,5 +67,6 @@ Arguments:
* `--model_dir`: Directory to save model training checkpoints. By default, it is `/tmp/ncf/`. * `--model_dir`: Directory to save model training checkpoints. By default, it is `/tmp/ncf/`.
* `--data_dir`: This should be set to the same directory given to the `data_download`'s `data_dir` argument. * `--data_dir`: This should be set to the same directory given to the `data_download`'s `data_dir` argument.
* `--dataset`: The dataset name to be downloaded and preprocessed. By default, it is `ml-1m`. * `--dataset`: The dataset name to be downloaded and preprocessed. By default, it is `ml-1m`.
* `--num_gpus`: The number of GPUs used for training/evaluation of the model. Use CPU if this flag is 0. By default, it is 1.
There are other arguments about models and training process. Refer to the [Flags package](https://abseil.io/docs/python/guides/flags) documentation or use the `--helpfull` flag to get a full list of possible arguments with detailed descriptions. There are other arguments about models and training process. Refer to the [Flags package](https://abseil.io/docs/python/guides/flags) documentation or use the `--helpfull` flag to get a full list of possible arguments with detailed descriptions.
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