latte

slurm_scripts/ffs.slurm

+#!/usr/bin/env bash
+#SBATCH --job-name=Latte-ffs # nom du job
+#SBATCH --partition group-name
+#SBATCH --nodes=1
+#SBATCH --ntasks-per-node=8
+#SBATCH --gres=gpu:8 # nombre de GPU �|  réserver (un unique GPU ici)
+#SBATCH --cpus-per-task=16
+#SBATCH --time=500:00:00 # temps exécution maximum demande (HH:MM:SS)
+#SBATCH --output=slurm_log/%j.out # nom du fichier de sortie
+#SBATCH --error=slurm_log/%j.err # nom du fichier d'erreur (ici commun avec la sortie)
+
+source ~/.bashrc
+
+conda activate latte
+
+srun python train.py --config ./configs/ffs/ffs_train.yaml
\ No newline at end of file

slurm_scripts/sky.slurm

+#!/usr/bin/env bash
+#SBATCH --job-name=Latte-ffs # nom du job
+#SBATCH --partition group-name
+#SBATCH --nodes=1
+#SBATCH --ntasks-per-node=8
+#SBATCH --gres=gpu:8 # nombre de GPU �|  réserver (un unique GPU ici)
+#SBATCH --cpus-per-task=16
+#SBATCH --time=500:00:00 # temps exécution maximum demande (HH:MM:SS)
+#SBATCH --output=slurm_log/%j.out # nom du fichier de sortie
+#SBATCH --error=slurm_log/%j.err # nom du fichier d'erreur (ici commun avec la sortie)
+
+source ~/.bashrc
+
+conda activate latte
+
+srun python train.py --config ./configs/sky/sky_train.yaml
\ No newline at end of file

train_scripts/ffs_train.sh

+export CUDA_VISIBLE_DEVICES=5
+# torchrun --nnodes=1 --nproc_per_node=2 --master_port=29509 train.py --config ./configs/ffs/ffs_train.yaml
+python train.py --config ./configs/ffs/ffs_train.yaml
\ No newline at end of file

train_scripts/sky_train.sh

+export CUDA_VISIBLE_DEVICES=4,5
+torchrun --nnodes=1 --nproc_per_node=2 --master_port=29509 train.py --config ./configs/sky/sky_train.yaml
\ No newline at end of file

train_scripts/taichi_train.sh

+export CUDA_VISIBLE_DEVICES=4,5
+torchrun --nnodes=1 --nproc_per_node=2 --master_port=29509 train.py --config ./configs/taichi/taichi_train.yaml
\ No newline at end of file