Omini3D / bash_infer.sh
maxmo2009's picture
Sync from local: code + epoch-110 checkpoint, clean README
2af0e94 verified
#!/bin/bash -l
#SBATCH --job-name=om-infer
#SBATCH --account=AIRR-P51-DAWN-GPU
#SBATCH --partition=pvc9
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH --time=10:00:00
#SBATCH --output=Logs/infer_%j.out
#SBATCH --error=Logs/infer_%j.err
. /etc/profile.d/modules.sh
module purge
module load rhel9/default-dawn
source ~/miniconda3/etc/profile.d/conda.sh
conda activate ~/rds/rds-airr-p51-TWhPgQVLKbA/Env/pub_env/pytorch-xpu
# python -u OM_aug.py -C Config/config_om.yaml
# python -u OM_reg.py -C Config/config_om.yaml
# python -u OM_reg_flexres.py -C Config/config_om.yaml
# python -u Scripts/OM_reg_flexres_om.py -C Config/config_om.yaml
# python -u Scripts/OM_reg_unpair.py -C Config/config_om.yaml -N 10
python -u Scripts/OM_reg_pair.py -C Config/config_reg_hip.yaml
# python -u OM_train_3modes.py -C Config/config_om.yaml --dummy-samples 20 --batchsize 1
# nohup python -u OM_aug.py -C Config/config_om.yaml > aug_log.txt 2>&1 &