Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Jiang Yu's avatar
    fix batch_size in transformer_main.py (#4897) · 2d7a0d6a
    Jiang Yu authored 
    * fix batch_size in transformer_main.py

fix batch_size in transformer_main.py which causes ResourceExhaustedError: OOM during training Transformer models using models/official/transformer

* small format change

change format from one line to multiple ones in order to pass lint tests

* remove trailing space and add comment
    2d7a0d6a
transformer_main.py 25.1 KB