Spaces:
Running
Running
File size: 1,888 Bytes
207ef6f |
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 |
from .tmux_launcher import Options, TmuxLauncher
class Launcher(TmuxLauncher):
def common_options(self):
return [
# Command 0
Options(
# NOTE: download the resized (and compressed) val set from
# http://efrosgans.eecs.berkeley.edu/CUT/datasets/cityscapes_val_for_CUT.tar
dataroot="datasets/cityscapes/cityscapes_val/",
direction="BtoA",
phase="val",
name="cityscapes_cut_pretrained",
CUT_mode="CUT",
),
# Command 1
Options(
dataroot="./datasets/cityscapes_unaligned/cityscapes/",
direction="BtoA",
name="cityscapes_fastcut_pretrained",
CUT_mode="FastCUT",
),
# Command 2
Options(
dataroot="./datasets/horse2zebra/",
name="horse2zebra_cut_pretrained",
CUT_mode="CUT"
),
# Command 3
Options(
dataroot="./datasets/horse2zebra/",
name="horse2zebra_fastcut_pretrained",
CUT_mode="FastCUT",
),
# Command 4
Options(
dataroot="/mnt/cloudNAS3/fangda/CycleGANData/dog2cat",
name="cat2dog_cut_pretrained",
CUT_mode="CUT"
),
# Command 5
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()]
|