|
from .tmux_launcher import Options, TmuxLauncher |
|
|
|
|
|
class Launcher(TmuxLauncher): |
|
def common_options(self): |
|
return [ |
|
|
|
Options( |
|
|
|
|
|
dataroot="datasets/cityscapes/cityscapes_val/", |
|
direction="BtoA", |
|
phase="val", |
|
name="cityscapes_cut_pretrained", |
|
CUT_mode="CUT", |
|
), |
|
|
|
|
|
Options( |
|
dataroot="./datasets/cityscapes_unaligned/cityscapes/", |
|
direction="BtoA", |
|
name="cityscapes_fastcut_pretrained", |
|
CUT_mode="FastCUT", |
|
), |
|
|
|
|
|
Options( |
|
dataroot="./datasets/horse2zebra/", |
|
name="horse2zebra_cut_pretrained", |
|
CUT_mode="CUT" |
|
), |
|
|
|
|
|
Options( |
|
dataroot="./datasets/horse2zebra/", |
|
name="horse2zebra_fastcut_pretrained", |
|
CUT_mode="FastCUT", |
|
), |
|
|
|
|
|
Options( |
|
dataroot="./datasets/afhq/cat2dog/", |
|
name="cat2dog_cut_pretrained", |
|
CUT_mode="CUT" |
|
), |
|
|
|
|
|
Options( |
|
dataroot="./datasets/afhq/cat2dog/", |
|
name="cat2dog_fastcut_pretrained", |
|
CUT_mode="FastCUT", |
|
), |
|
|
|
|
|
] |
|
|
|
def commands(self): |
|
return ["python train.py " + str(opt) for opt in self.common_options()] |
|
|
|
def test_commands(self): |
|
return ["python test.py " + str(opt.set(num_test=500)) for opt in self.common_options()] |
|
|