#!/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)"