1. 04 Nov, 2020 7 commits
  2. 03 Nov, 2020 14 commits
  3. 02 Nov, 2020 15 commits
  4. 01 Nov, 2020 2 commits
  5. 30 Oct, 2020 2 commits
    • Abi See's avatar
      Fix two bugs with --logging_first_step (#8193) · 8f1c960e
      Abi See authored
      * make sure that logging_first_step evaluates
      
      * fix bug with incorrect loss on logging_first_step
      
      * fix style
      
      * logging_first_step only logs, not evals
      8f1c960e
    • Avital Oliver's avatar
      Minor style improvements for the Flax BERT and RoBERTa examples (#8178) · 689ff74f
      Avital Oliver authored
      * Minor style improvements:
      
      1. Use `@nn.compact` rather than `@compact` (as to not make it seem
         like compact is a standard Python decorator.
      2. Move attribute docstrings from two `__call__` methods to comments
         on the attributes themselves. (This was probably a remnant from
         the pre-Linen version where the attributes were arguments to
         `call`.)
      
      * Use black on the Flax modeling code
      689ff74f