Load and print networks; create schedulers Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseOptions
(self, opt)
| 75 | pass |
| 76 | |
| 77 | def setup(self, opt): |
| 78 | """Load and print networks; create schedulers |
| 79 | |
| 80 | Parameters: |
| 81 | opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseOptions |
| 82 | """ |
| 83 | if self.isTrain: |
| 84 | self.schedulers = [networks.get_scheduler(optimizer, opt) for optimizer in self.optimizers] |
| 85 | if not self.isTrain or opt.continue_train: |
| 86 | load_suffix = '%d' % opt.load_iter if opt.load_iter > 0 else opt.epoch |
| 87 | self.load_networks(load_suffix) |
| 88 | self.print_networks(opt.verbose) |
| 89 | |
| 90 | def eval(self): |
| 91 | """Make models eval mode during test time""" |
no test coverage detected