Browse Source

单卡

master
liupengfei 1 month ago
parent
commit
6ba5a36049
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      default_config.yaml

+ 1
- 1
default_config.yaml View File

@@ -27,7 +27,7 @@ train_data: "FlyingChairs" # Train Dataset name
train_data_path: "/home/work/user-job-dir/data/FlyingChairs_part" # Train Dataset path

# Train Setup
run_distribute: 1 # Distributed training or not
run_distribute: 0 # Distributed training or not
is_save_on_master: 1 # Only save ckpt on master device
save_checkpoint: 1 # Is save ckpt while training
save_ckpt_interval: 2 # Saving ckpt interval


Loading…
Cancel
Save