squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #110 from eltociear/patch-1

Browse Source 
fix typo in default.yaml
This commit is contained in:
Tri Dao 2023-01-27 12:18:16 -08:00 committed by GitHub
commit 06da275bcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training/configs/callbacks/default.yaml
View File

@ -9,7 +9,7 @@ model_checkpoint:
monitor: "val/acc" # name of the logged metric which determines when model is improving monitor: "val/acc" # name of the logged metric which determines when model is improving
mode: "max" # can be "max" or "min" mode: "max" # can be "max" or "min"
save_top_k: 1 # save k best models (determined by above metric) save_top_k: 1 # save k best models (determined by above metric)
save_last: True # additionaly always save model from last epoch save_last: True # additionally always save model from last epoch
verbose: False verbose: False
dirpath: ${oc.env:CHECKPOINT_DIR,checkpoints}/${oc.select:name,''} dirpath: ${oc.env:CHECKPOINT_DIR,checkpoints}/${oc.select:name,''}
filename: "epoch_{epoch:03d}" filename: "epoch_{epoch:03d}"