Upload folder using huggingface_hub

30e9297 verified 5 months ago

229 Bytes

#!/bin/bash
# Train the music generation model
set -e
cd "$(dirname "$0")/.."

python -m src.s00_main train \
    --epochs ${EPOCHS:-50} \
    --batch-size ${BATCH_SIZE:-8} \
    --lr ${LR:-3e-4} \
    --seq-len ${SEQ_LEN:-1024}

	#!/bin/bash
	# Train the music generation model
	set -e
	cd "$(dirname "$0")/.."

	python -m src.s00_main train \
	--epochs ${EPOCHS:-50} \
	--batch-size ${BATCH_SIZE:-8} \
	--lr ${LR:-3e-4} \
	--seq-len ${SEQ_LEN:-1024}