Spaces:
Sleeping
Sleeping
File size: 1,260 Bytes
e75a247 |
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 |
#!/bin/bash
sbatch <<EOT
#!/bin/bash
#SBATCH --partition=gpu # Specify the partition
#SBATCH --mem=25000 # Request 10GB of memory
#SBATCH --time=40:00:00
#SBATCH --job-name=SVJTransformerTraining # Name the job
#SBATCH --gres=gpu:1
#SBATCH --account=gpu_gres # Specify the account
#SBATCH --output=jobs/vega/training_transformer_out_$1_$2_$3.log
#SBATCH --error=jobs/vega/training_transformer_err_$1_$2_$3.log
source env.sh
export APPTAINER_TMPDIR=/work/gkrzmanc/singularity_tmp
export APPTAINER_CACHEDIR=/work/gkrzmanc/singularity_cache
nvidia-smi
srun singularity exec -B /t3home/gkrzmanc/ -B /work/gkrzmanc/ --nv docker://gkrz/lgatr:v3 python -m src.train -train scouting_PFNano_signals2/SVJ_hadronic_std3/s-channel_mMed-900_mDark-20_rinv-0.3_alpha-peak_13TeV-pythia8_n-2000 -val scouting_PFNano_signals2/SVJ_hadronic_std2/s-channel_mMed-900_mDark-20_rinv-0.3 -net src/models/transformer/transformer.py -bs 64 --gpus 0 --run-name Transformer_training_40k_$1_$2_$3 --val-dataset-size 1024 --num-steps 40000 --attr-loss-weight 0.1 --coord-loss-weight 0.1 --beta-type pt+bc --gt-radius 0.8 --num-blocks $1 --internal-dim $2 --n-heads $3
exit 0
EOT
# Args: n_blocks internal_dim n_heads (default: 10, 128, 4)
|