Nanobit commited on
Commit
48c5647
·
unverified ·
1 Parent(s): 36b2e1c

Fix(doc): Clarify no amp to full yaml docs (#496)

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -407,6 +407,10 @@ fp16: true
407
  # Use CUDA tf32
408
  tf32: true # require >=ampere
409
 
 
 
 
 
410
  # a list of one or more datasets to finetune the model with
411
  datasets:
412
  # hf dataset repo | "json" for local dataset, make sure to fill data_files
 
407
  # Use CUDA tf32
408
  tf32: true # require >=ampere
409
 
410
+ # No AMP (automatic mixed precision)
411
+ bfloat16: true # require >=ampere
412
+ float16: true
413
+
414
  # a list of one or more datasets to finetune the model with
415
  datasets:
416
  # hf dataset repo | "json" for local dataset, make sure to fill data_files