-
Notifications
You must be signed in to change notification settings - Fork 0
/
slurm_launcher.sh
83 lines (72 loc) · 4.08 KB
/
slurm_launcher.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/bin/bash
#SBATCH --job-name=te-stargan-lh # nom du job
#SBATCH --ntasks=1 # number of MP tasks
#SBATCH --ntasks-per-node=1 # number of MPI tasks per node
#SBATCH --partition=gpu_p13
#SBATCH --qos=qos_gpu-dev
#SBATCH --gres=gpu:1 # number of GPUs per node
#SBATCH --cpus-per-task=16 # number of cores per tasks
#SBATCH --hint=nomultithread # we get physical cores not logical
#SBATCH --distribution=block:block # we pin the tasks on contiguous cores
#SBATCH --time=2:00:00 # maximum execution time (HH:MM:SS)
#SBATCH --output=te-stargan-lh%j.out # output file name
#SBATCH --error=te-stargan-lh%j.err # error file name
source /gpfswork/rech/gft/umh25bv/miniconda3/bin/activate /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv
# Feature extractor
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style-transfer_diffusion/main.py \
# --model feature_extractor --data_dir data \
# --dataset dataset_rh_4classes-jeanzay --labels pipelines \
# --model_save_dir feature_extractor/models --batch_size 64 \
# --lrate 1e-4 --n_epoch 150
# Pix2Pix / CycleGAN
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style-transfer_diffusion/main.py \
# --model cycleGAN --mode train --dataset dataset_rh_4classes-jeanzay \
# --labels pipelines --image_size 56 --c_dim 4 --batch_size 16 \
# --data_dir data --sample_dir results/samples/cycleGAN-rh \
# --model_save_dir results/models/cycleGAN-rh
# StarGAN
## Train
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style-transfer_diffusion/main.py \
# --model stargan --mode train --dataset dataset_lh_4classes-jeanzay \
# --labels pipelines --image_size 56 --c_dim 4 --batch_size 16 \
# --data_dir data --sample_dir results/samples/stargan-lh \
# --model_save_dir results/models/stargan-lh
# Test
/gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style-transfer_diffusion/main.py \
--model stargan --mode test --dataset dataset_lh_4classes-jeanzay \
--labels pipelines --image_size 56 --c_dim 4 --batch_size 1 \
--data_dir data --sample_dir results/samples/stargan-lh--lh \
--model_save_dir results/models/stargan-lh --test_iter 60000
#C-DDPM
## Train
## Test
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style_transfer/main.py \
# --model c_ddpm --mode transfer --dataset dataset_rh_4classes-jeanzay \
# --labels pipelines --batch_size 1 --data_dir data --sample_dir c_ddpm/samples \
# --model_save_dir c_ddpm/models --test_iter 90 --n_classes 4
# CC-DDPM
## Train
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style-transfer_diffusion/main.py \
# --model cc_ddpm --mode train --dataset dataset_rh_4classes-jeanzay \
# --labels pipelines --model_save_dir cc_ddpm/models \
# --batch_size 8 --lrate 1e-4 --n_epoch 200 --n_classes 4 \
# --sample_dir cc_ddpm/samples
## Test
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style-transfer_diffusion/main.py \
# --model cc_ddpm --mode transfer --dataset dataset_rh_4classes-jeanzay \
# --labels pipelines --model_save_dir results/models/cc_ddpm-rh \
# --batch_size 1 --n_classes 4 --sampling random --n_C 1 \
# --sample_dir results/samples/cc_ddpm-rh--rh --test_iter 190
# CC-DDPM-S
## Train
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style_transfer/main.py \
# --model cc_ddpm_s --mode train --dataset dataset_rh_4classes-jeanzay \
# --labels pipelines --model_save_dir cc_ddpm_s/models \
# --batch_size 8 --lrate 1e-4 --n_epoch 200 --n_classes 4 \
# --sample_dir cc_ddpm_s/samples
## Test
# /gpfswork/rech/gft/umh25bv/miniconda3/envs/workEnv/bin/python3 -u /gpfswork/rech/gft/umh25bv/style_transfer/main.py \
# --model cc_ddpm_s --mode transfer --dataset dataset_rh_4classes-jeanzay \
# --labels pipelines --model_save_dir cc_ddpm_s/models \
# --batch_size 1 --n_classes 4 \
# --sample_dir cc_ddpm_s/samples --test_iter 190