xfcghj
/

AR

Model card Files Files and versions

AR / mesh /scripts /train.sh

xfcghj's picture

Upload folder using huggingface_hub

f0fc238 verified 2 months ago

History Blame Contribute Delete

194 Bytes

	#!/bin/bash

	export CUDA_VISIBLE_DEVICES=4,5,6,7

	torchrun \
	--nproc_per_node=4 \
	--master_port=29889 \
	train.py \
	--batch_size 2 \
	--epochs 100 \
	--save_every_k_epochs 1