Startseite Erkunden Hilfe
Registrieren Anmelden
david
/
flash-attention
Mirror von https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Dateien Issues 0 Wiki
Struktur: 0bf5e50038
Branches Tags
flash-attention
/
training
/
configs
/
optimizer
/
adamw-apex.yaml

adamw-apex.yaml 81 B
Verlauf Originalformat

123 
  1. # @package train.optimizer
    2. 

  2. _target_: apex.optimizers.FusedAdam
    3. 

  3. adam_w_mode: True
    4.