distributed_type: DEEPSPEED
deepspeed_config:
  zero_stage: 3
num_processes: 2