Files
2025-06-11 12:32:04 +00:00

5 lines
150 B
Bash
Executable File

#!/bin/bash
torchrun --nnodes=1 --nproc_per_node=1 --master_port=25679 -m scripts.train_unet \
--unet_config_path "configs/unet/stage1_512.yaml"