File size: 1,614 Bytes
58e7af4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
#!/bin/bash
#SBATCH -J down_on # Job name
#SBATCH -o down_on_%j.out # Name of stdout output log file (%j expands to jobID)
#SBATCH -e down_on_%j.err # Name of stderr output log file (%j expands to jobID)
#SBATCH -N 1 # Total number of nodes requested
#SBATCH -n 1 # Total number of cores requested
#SBATCH --mem=5000 # Total amount of (real) memory requested (per node)
#SBATCH -t 10:00:00 # Time limit (hh:mm:ss)
#SBATCH --partition=gpu # Request partition for resource allocation
#SBATCH --gres=gpu:1 # Specify a list of generic consumable resources (per node)
cd /home/mr249/ac_h
/home/mr249/env/hugh/bin/python run_audio_classification.py \
--model_name_or_path facebook/wav2vec2-base \
--dataset_name MatsRooth/down_on \
--output_dir wav2vec2-base_down_on \
--overwrite_output_dir \
--remove_unused_columns False \
--do_train \
--do_eval \
--fp16 \
--learning_rate 3e-5 \
--max_length_seconds 1 \
--attention_mask False \
--warmup_ratio 0.1 \
--num_train_epochs 5 \
--per_device_train_batch_size 32 \
--gradient_accumulation_steps 4 \
--per_device_eval_batch_size 32 \
--dataloader_num_workers 1 \
--logging_strategy steps \
--logging_steps 10 \
--evaluation_strategy epoch \
--save_strategy epoch \
--load_best_model_at_end True \
--metric_for_best_model accuracy \
--save_total_limit 3 \
--seed 0
|