Skip to content

Updating bfloat16

Latest
Compare
Choose a tag to compare
@layumi layumi released this 07 May 08:29
  • Change Nvidia apex to the native pytorch
  • Support bfloat16 (by --bf16)
  • Fix the USAM error
  • Adding DINO style loss and EMA for --DG