Spaces:
Running
Running
File size: 1,605 Bytes
1fd7780 |
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 |
from argparse import ArgumentParser
class TestOptions:
def __init__(self):
self.parser = ArgumentParser()
self.initialize()
def initialize(self):
# arguments for inference script
self.parser.add_argument('--exp_dir', type=str, help='Path to experiment output directory')
self.parser.add_argument('--checkpoint_path', default=None, type=str, help='Path to model checkpoint')
self.parser.add_argument('--couple_outputs', action='store_true', help='Whether to also save inputs + outputs side-by-side')
self.parser.add_argument('--mapper_type', default='LevelsMapper', type=str, help='Which mapper to use')
self.parser.add_argument('--no_coarse_mapper', default=False, action="store_true")
self.parser.add_argument('--no_medium_mapper', default=False, action="store_true")
self.parser.add_argument('--no_fine_mapper', default=False, action="store_true")
self.parser.add_argument('--stylegan_size', default=1024, type=int)
self.parser.add_argument('--test_batch_size', default=2, type=int, help='Batch size for testing and inference')
self.parser.add_argument('--latents_test_path', default=None, type=str, help="The latents for the validation")
self.parser.add_argument('--test_workers', default=1, type=int, help='Number of test/inference dataloader workers')
self.parser.add_argument('--work_in_stylespace', default=False, action='store_true')
self.parser.add_argument('--n_images', type=int, default=None, help='Number of images to output. If None, run on all data')
def parse(self):
opts = self.parser.parse_args()
return opts |