|
from .tmux_launcher import Options, TmuxLauncher |
|
|
|
|
|
class Launcher(TmuxLauncher): |
|
def common_options(self): |
|
return [ |
|
|
|
Options( |
|
dataroot="./datasets/grumpifycat", |
|
name="grumpifycat_CUT", |
|
CUT_mode="CUT" |
|
), |
|
|
|
|
|
Options( |
|
dataroot="./datasets/grumpifycat", |
|
name="grumpifycat_FastCUT", |
|
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(phase='train')) for opt in self.common_options()] |
|
|