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
|