YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add train_config.yaml

Browse Source
This commit is contained in:
三洋三洋
2024-05-07 07:11:05 +00:00
parent 3c0ec95b26
commit ef529f9234
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/train_config.yaml Normal file
View File

@@ -0,0 +1,14 @@
compute_environment: LOCAL_MACHINE
debug: false
distributed_type: MULTI_GPU
gpu_ids: all
num_processes: 1
machine_rank: 0
main_training_function: main
num_machines: 1
rdzv_backend: static
same_network: true
tpu_env: []
tpu_use_cluster: false
tpu_use_sudo: false
use_cpu: false