File size: 1,060 Bytes
c1a7f73 |
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 |
#!/bin/bash
mkdir -p job_out
#SBATCH --job-name YOUR_JOB_NAME # Job name
### Logging
#SBATCH --output=job_out/%j.out # Stdout (%j expands to jobId)
#SBATCH --error=job_out/%j.err # Stderr (%j expands to jobId)
### Node info
#SBATCH --nodes=1 # Single node or multi node
#SBATCH --nodelist=sota-6
#SBATCH --time 20:00:00 # Max time (hh:mm:ss)
#SBATCH --gres=gpu:4 # GPUs per node
#SBATCH --mem=256G # Recommend 32G per GPU
#SBATCH --ntasks-per-node=4 # Tasks per node
#SBATCH --cpus-per-task=256 # Recommend 8 per GPU
### Whatever your job needs to do
export REQUESTS_CA_BUNDLE="/etc/ssl/certs/ca-certificates.crt"
export HTTPS_PROXY="https://192.168.0.10:443/"
export https_proxy="https://192.168.0.10:443/"
export TEST_VAL_PRED=True
export WANDB=True
cd /u/xiuyu/work/dev4
PYTHONPATH=".":$PYTHONPATH python3 train.py --config configs/train/pretrain_scalable_map.yaml --save_ckpt_path output/ours_map_pretrain
|