run_dp_lr_onehot.sh

#!/bin/bash -l

#SBATCH -n 4
#SBATCH --cpus-per-task=4
#SBATCH --time=24:00:00
#SBATCH --mem-per-cpu=4096
#SBATCH --gpus=1
#SBATCH --gres=gpumem:20000m

module load eth_proxy
module load gcc/9.3.0
module load cuda/11.7.0
conda activate PvP
wandb login

export TASK_NAME=ner
export CUDA_LAUNCH_BLOCKING=1

python3 run_dp.py \
  --do_train \
  --do_eval \
  --per_device_train_batch_size 32 \
  --per_device_eval_batch_size 32 \
  --gpt2_name_or_path gpt2 \
  --data_dir ontonotes/dp/ \
  --task $TASK_NAME \
  --output_dir outputs/dp/lr/$TASK_NAME/onehot \
  --overwrite_output_dir \
  --cache_dir cache/ \
  --mlp_dropout 0.0 \
  --num_train_epochs 256.0 \
  --learning_rate 1e-6 \
  --weight_decay 0.0 \
  --mlp_dim 512 \
  --mlp_layers 128 \
  --fp16 \
  --use_mlp False \
  --dev \
  --save_strategy epoch \
  --evaluation_strategy epoch \
  --randomized \
  --onehot True \