7 lines
119 B
YAML
7 lines
119 B
YAML
amp_enabled: false
|
|
amp_device: "cuda"
|
|
cuda: true
|
|
amp_dtype: f32
|
|
torch_deterministic: false
|
|
device_rank: 0
|
|
compile: true |