From 6203e75fe4baebba8545a0a4081fa1ccf73dc2e1 Mon Sep 17 00:00:00 2001 From: Guillaume-Duret <guillaume.duret@ec-lyon.fr> Date: Fri, 2 Jun 2023 15:30:46 +0200 Subject: [PATCH] slurm eval --- datasets/linemod/dataset.py | 2 +- slurm/DF_eval.slurm | 27 +++++++++++++++++++++++++++ slurm/DF_train.slurm | 31 +++++++++++++++++++++++++++++++ 3 files changed, 59 insertions(+), 1 deletion(-) create mode 100644 slurm/DF_eval.slurm create mode 100644 slurm/DF_train.slurm diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py index f060ac3..86d7a8f 100755 --- a/datasets/linemod/dataset.py +++ b/datasets/linemod/dataset.py @@ -61,7 +61,7 @@ class PoseDataset(data.Dataset): self.list_rgb.append('{0}/data/{1}/rgb/{2}.png'.format(self.root, '%02d' % item, input_line)) self.list_depth.append('{0}/data/{1}/depth/{2}.png'.format(self.root, '%02d' % item, input_line)) if self.mode == 'eval': - self.list_label.append('{0}/segnet_results_gt/{1}_label/{2}.png'.format(self.root, '%02d' % item, input_line)) + self.list_label.append('{0}/segnet_results/{1}_label_gt/{2}.png'.format(self.root, '%02d' % item, input_line)) else: self.list_label.append('{0}/data/{1}/mask/{2}.png'.format(self.root, '%02d' % item, input_line)) diff --git a/slurm/DF_eval.slurm b/slurm/DF_eval.slurm new file mode 100644 index 0000000..8263796 --- /dev/null +++ b/slurm/DF_eval.slurm @@ -0,0 +1,27 @@ +#!/bin/bash +#SBATCH --job-name=DF07cam # nom du job +#SBATCH -A uli@gpu +#SBATCH --ntasks=1 # Nombre de processus MPI +#SBATCH --cpus-per-task=10 # nombre de threads OpenMP +#SBATCH --hint=nomultithread # pas d'hyperthreading +#SBATCH --qos=qos_gpu-t4 +#SBATCH --gres=gpu:1 +#SBATCH --time=100:00:00 # Temps d’exécution max +#SBATCH --output=Output/Pvnet%j.out # fichier de sortie +#SBATCH --error=Output/Pvnet%j.out # fichier d'erreur + +# on se place dans le répertoire de soumission +cd ${SLURM_SUBMIT_DIR} + +module purge +# chargement des modules + +module load python/3.6.15 + +# echo des commandes lancées +set -x + + +conda activate NewDF +bash ./experiments/scripts/eval_linemod.sh + diff --git a/slurm/DF_train.slurm b/slurm/DF_train.slurm new file mode 100644 index 0000000..d43ec06 --- /dev/null +++ b/slurm/DF_train.slurm @@ -0,0 +1,31 @@ +#!/bin/bash +#SBATCH --job-name=DF07cam # nom du job +#SBATCH -A uli@gpu +#SBATCH --ntasks=1 # Nombre de processus MPI +#SBATCH --cpus-per-task=10 # nombre de threads OpenMP +#SBATCH --hint=nomultithread # pas d'hyperthreading +#SBATCH --qos=qos_gpu-t4 +#SBATCH --gres=gpu:1 +#SBATCH --time=100:00:00 # Temps d’exécution max +#SBATCH --output=Output/Pvnet%j.out # fichier de sortie +#SBATCH --error=Output/Pvnet%j.out # fichier d'erreur + +# on se place dans le répertoire de soumission +cd ${SLURM_SUBMIT_DIR} + +module purge +# chargement des modules + +module load python/3.6.15 + +# echo des commandes lancées +set -x + + +cat_target=$1 +echo $cat_target + + +conda activate NewDF +bash ./experiments/scripts/train_linemod.sh + -- GitLab