train_model_cluster_imdb.sh

#!/bin/bash

module load gcc/8.2.0 python_gpu/3.10.4 eth_proxy
source ../pp_env_tf_python310/bin/activate

sbatch --mem-per-cpu=12g\
       --gpus=1\
       --gres=gpumem:12g\
       --time=30:00:00\
       --wrap "python ../moderation_classifier/train_BERT.py"