1. 16 Oct, 2019 3 commits
    • Yeqing Li's avatar
      Enables timer callback and disables checkpoint saving in retinanet benchmark test. · 1b77cd80
      Yeqing Li authored
      PiperOrigin-RevId: 275080469
      1b77cd80
    • 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
    • Yeqing Li's avatar
      Internal change · e97e22df
      Yeqing Li authored
      PiperOrigin-RevId: 274921478
      e97e22df
  2. 15 Oct, 2019 6 commits
  3. 14 Oct, 2019 1 commit
  4. 13 Oct, 2019 2 commits
  5. 12 Oct, 2019 2 commits
  6. 11 Oct, 2019 8 commits
  7. 10 Oct, 2019 8 commits
  8. 09 Oct, 2019 3 commits
  9. 08 Oct, 2019 2 commits
  10. 07 Oct, 2019 4 commits
  11. 05 Oct, 2019 1 commit