Spaces:

nishu08
/

sql-error-classifier-train

Runtime error

sql-error-classifier-train / scripts /run_codebert_training.sh

Deploy CodeBERT training Space

9b2cded verified 5 months ago

390 Bytes

#!/usr/bin/env bash
set -euo pipefail

DATA="${1:-data/sql_errors_dev.parquet}"
OUTPUT="${2:-models/codebert-cross-encoder}"
SAMPLES="${3:-}"

CMD=(python -m src.hf_train_codebert --data "${DATA}" --output-dir "${OUTPUT}")

if [[ -n "${SAMPLES}" ]]; then
  CMD+=(--max-samples "${SAMPLES}")
fi

echo "==> Training CodeBERT cross-encoder..."
"${CMD[@]}"

echo "==> Done. Model at ${OUTPUT}"

	#!/usr/bin/env bash
	set -euo pipefail

	DATA="${1:-data/sql_errors_dev.parquet}"
	OUTPUT="${2:-models/codebert-cross-encoder}"
	SAMPLES="${3:-}"

	CMD=(python -m src.hf_train_codebert --data "${DATA}" --output-dir "${OUTPUT}")

	if [[ -n "${SAMPLES}" ]]; then
	CMD+=(--max-samples "${SAMPLES}")
	fi

	echo "==> Training CodeBERT cross-encoder..."
	"${CMD[@]}"

	echo "==> Done. Model at ${OUTPUT}"