Fix(doc): Clarify no amp to full yaml docs (#496)
Browse files
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
|