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