• Madhura Pande's avatar
    Tensorboard summary writing for few params · 1f4936f1
    Madhura Pande authored
    Using this code on GPU, was not logging "learning_rate" and "gradient norm" etc. We need to use simple "tf.summary.scalar" instead of contrib one to log these params. Tested on Tensorflow 1.14.
    1f4936f1
transformer_main.py 26.6 KB