-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdn.run
36 lines (34 loc) · 862 Bytes
/
dn.run
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
#!/usr/bin/bash
export OMP_NUM_THREADS=1
for SEED in 2020 2021 2022
do
torchrun --nproc_per_node=8 experiment.py \
--batch_size 50 \
--path /dataB3/tommie_kerssies \
--method cotta \
--seed ${SEED} \
--epochs 1 \
--model best_real_${SEED}.pth.tar \
--sources real \
--targets painting clipart sketch \
--dataset DomainNet-126 \
--lr 1e-3 \
--rst_m 0.1 \
--mt_alpha 0.99
done
for SEED in 2020 2021 2022
do
torchrun --nproc_per_node=8 experiment.py \
--batch_size 50 \
--path /dataB3/tommie_kerssies \
--method cotta \
--seed ${SEED} \
--epochs 1 \
--model best_painting_${SEED}.pth.tar \
--sources painting \
--targets real clipart sketch \
--dataset DomainNet-126 \
--lr 1e-3 \
--rst_m 0.1 \
--mt_alpha 0.99
done