23 lines
425 B
Bash
23 lines
425 B
Bash
#!/bin/bash
|
|
|
|
#SBATCH --job-name=afem-train
|
|
#SBATCH --partition=gpu
|
|
#SBATCH --gres=gpu:1
|
|
#SBATCH --nodelist=node06
|
|
#SBATCH --cpus-per-task=4
|
|
#SBATCH --mem=32G
|
|
#SBATCH --time=24:00:00
|
|
#SBATCH --output=logs/train_%j.out
|
|
|
|
# cd /public/home/dxw/Codes/afem
|
|
|
|
|
|
|
|
echo "Starting training at $(date)"
|
|
|
|
echo "Running on node: $(hostname)"
|
|
|
|
python -u src/main.py --mode train --config src/config.yaml
|
|
|
|
echo "Training finished at $(date)"
|