(self)
| 2 | |
| 3 | class BaseOptions(): |
| 4 | def parse(self): |
| 5 | parser = argparse.ArgumentParser() |
| 6 | self.parser = self.initialize(parser) |
| 7 | self.opt = self.parser.parse_args() |
| 8 | return self.opt |
| 9 | |
| 10 | def initialize(self, parser): |
| 11 | parser.add_argument('--pretrained_dir', type=str, default='./checkpoints') |