Skip to content

GitLab

Switch branch/tag
  1. 05 Mar, 2020 1 commit
  3. 03 Dec, 2019 1 commit
  5. 26 Nov, 2019 1 commit
  7. 16 Oct, 2019 1 commit
    • Reed Wanderman-Milne's avatar
      Add support for the tf.keras.mixed_precision API in NCF · cb913691
      Reed Wanderman-Milne authored 
      To test, I did 50 fp32 runs and 50 fp16 runs. I used the following command:

python ncf_keras_main.py --dataset=ml-20m --num_gpus=1 --train_epochs=10 --clean --batch_size=99000 --learning_rate=0.00382059 --beta1=0.783529 --beta2=0.909003 --epsilon=1.45439e-7 --layers=256,256,128,64 --num_factors=64 --hr_threshold=0.635 --ml_perf --nouse_synthetic_data --data_dir ~/ncf_data_dir_python3 --model_dir ~/tmp_model_dir --keras_use_ctl

For the fp16 runs, I added --dtype=fp16. The average hit-rate for both fp16 and fp32 was 0.6365. I also did 50 runs with the mixed precision graph rewrite, and the average hit-rate was 0.6363. The difference is likely due to noise.

PiperOrigin-RevId: 275059871
      cb913691
  9. 24 Sep, 2019 1 commit
  11. 17 Sep, 2019 1 commit
  13. 05 Sep, 2019 1 commit
  15. 04 Sep, 2019 1 commit
  17. 14 Aug, 2019 4 commits
  19. 12 Aug, 2019 1 commit
    • Hongjun Choi's avatar
      Merged commit includes the following changes: (#7430) · 03b4a0af
      Hongjun Choi authored 
      262988559  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Enable NCF TF 2.0 model to run on TPUStrategy.

--
262971756  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Internal change

262967691  by hongkuny<hongkuny@google.com>:

    Internal

--

PiperOrigin-RevId: 262988559
      03b4a0af
  21. 09 Aug, 2019 2 commits
  23. 06 Aug, 2019 1 commit
  25. 31 Jul, 2019 2 commits
    • Hongjun Choi's avatar
      Merged commit includes the following changes: (#7346) · 1b089751
      Hongjun Choi authored 
      * Merged commit includes the following changes:
260800447  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Fix tf function signature errors for NCF model.

--
260767064  by psv<psv@google.com>:

    Renames the `run_distributed` flag to `experimental_run_tf_function` and modifies the tests generated by the keras_all_modes decorator to better reflect that.

--

PiperOrigin-RevId: 260800447

* Update ncf_keras_benchmark.py

* Update ncf_keras_benchmark.py

* Update ncf_keras_benchmark.py

* Update ncf_keras_benchmark.py
      1b089751
    • Toby Boyd's avatar
      Change to experimental_run_tf_function. (#7344) · a552e76a
      Toby Boyd authored
      a552e76a
  27. 30 Jul, 2019 1 commit
  29. 29 Jul, 2019 1 commit
    • Hongjun Choi's avatar
      Merged commit includes the following changes: (#7322) · 803f833c
      Hongjun Choi authored 
      260228553  by priyag<priyag@google.com>:

    Enable transformer and NCF official model tests. Also fix some minor issues so that all tests pass with TF 1 + enable_v2_behavior.

--
260043210  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Add logic to train NCF model using offline generated data.

--
259778607  by priyag<priyag@google.com>:

    Internal change

259656389  by hongkuny<hongkuny@google.com>:

    Internal change

PiperOrigin-RevId: 260228553
      803f833c
  31. 23 Jul, 2019 1 commit
  33. 20 Jul, 2019 1 commit
  35. 19 Jul, 2019 2 commits
  37. 16 Jul, 2019 1 commit
  39. 08 Jul, 2019 1 commit
  41. 28 Jun, 2019 1 commit
  43. 21 Jun, 2019 1 commit
    • Toby Boyd's avatar
      NCF XLA and Eager tests with a refactor of resnet flags to make this cleaner. (#7067) · a68f65f8
      Toby Boyd authored 
      * XLA FP32 and first test

* More XLA benchmarks FP32.

* Add eager to NCF and refactor resnet.

* fix v2_0 calls and more flag refactor.

* Remove extra flag args.

* 90 epoch default

* add return

* remove xla not used by estimator.

* Remove duplicate run_eagerly.

* fix flag defaults.

* Remove fp16_implementation flag option.

* Remove stop early on mlperf test.

* remove unneeded args.

* load flags from keras mains.
      a68f65f8
  45. 13 Jun, 2019 7 commits
  47. 05 Jun, 2019 4 commits
  49. 03 Jun, 2019 1 commit