lsn-analysis / job_scripts /activation.sh
tvkain's picture
Upload folder using huggingface_hub
fed1832 verified
#!/bin/bash
#SBATCH --job-name=activation
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=8
#SBATCH --partition=physical-gpu
#SBATCH --mem=32G
#SBATCH --output=logs/%j-activation.out
ENV_FILE=$1
if [ -z "$ENV_FILE" ]; then
echo "Error: No environment file specified"
exit 1
fi
if [ ! -f "$ENV_FILE" ]; then
echo "Error: Environment file $ENV_FILE not found"
exit 1
fi
# Load the specified environment file
source "$ENV_FILE"
# Create the artifact folder
mkdir -p "$OUTPUT_DIR"
export PYTHONUNBUFFERED=1
# Run the Python script
python ../activation.py \
--model $MODEL \
--output-dir $OUTPUT_DIR \
--data-path $DATA_PATH \
--lang $LANG
echo "Activation single job done"