Skip to content

GitLab

Switch branch/tag
  1. 29 Apr, 2020 1 commit
  3. 25 Apr, 2020 1 commit
  5. 22 Apr, 2020 1 commit
  7. 16 Apr, 2020 1 commit
  9. 14 Apr, 2020 1 commit
  11. 10 Apr, 2020 1 commit
  13. 09 Apr, 2020 2 commits
  15. 30 Mar, 2020 1 commit
  17. 17 Mar, 2020 1 commit
  19. 14 Mar, 2020 1 commit
  21. 11 Mar, 2020 1 commit
  23. 05 Mar, 2020 1 commit
  25. 02 Mar, 2020 1 commit
  27. 28 Feb, 2020 1 commit
  29. 25 Feb, 2020 1 commit
  31. 21 Feb, 2020 2 commits
  33. 20 Feb, 2020 1 commit
  35. 13 Feb, 2020 1 commit
  37. 29 Jan, 2020 1 commit
  39. 27 Jan, 2020 1 commit
  41. 21 Jan, 2020 1 commit
  43. 17 Jan, 2020 1 commit
  45. 15 Dec, 2019 1 commit
  47. 14 Dec, 2019 2 commits
  49. 11 Dec, 2019 1 commit
  51. 06 Dec, 2019 1 commit
  53. 05 Dec, 2019 1 commit
  55. 04 Dec, 2019 1 commit
  57. 27 Nov, 2019 1 commit
  59. 25 Nov, 2019 1 commit
  61. 21 Nov, 2019 1 commit
  63. 19 Nov, 2019 1 commit
  65. 18 Nov, 2019 1 commit
  67. 11 Nov, 2019 1 commit
  69. 28 Oct, 2019 1 commit
  71. 21 Oct, 2019 1 commit
  73. 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