Spaces:

maxin-cn
/

Latte-1

Running on Zero

Latte-1 / train_scripts /taichi_train.sh

Upload folder using huggingface_hub

94bafa8 verified 3 months ago

143 Bytes

	export CUDA_VISIBLE_DEVICES=4,5
	torchrun --nnodes=1 --nproc_per_node=2 --master_port=29509 train.py --config ./configs/taichi/taichi_train.yaml