mirror of
https://github.com/GoogleCloudPlatform/vertex-ai-samples.git
synced 2026-09-26 14:42:04 +00:00
7 lines
191 B
Bash
7 lines
191 B
Bash
#!/bin/bash
|
|
|
|
# Setup accelerate config before running trainer.
|
|
python -c "from accelerate.utils import write_basic_config; write_basic_config(mixed_precision='fp16')"
|
|
|
|
accelerate launch "$@"
|