(self)
| 9 | |
| 10 | class Train: |
| 11 | def __init__(self): |
| 12 | self._opt = TrainOptions().parse() |
| 13 | data_loader_train = CustomDatasetDataLoader(self._opt, is_for_train=True) |
| 14 | data_loader_test = CustomDatasetDataLoader(self._opt, is_for_train=False) |
| 15 | |
| 16 | self._dataset_train = data_loader_train.load_data() |
| 17 | self._dataset_test = data_loader_test.load_data() |
| 18 | |
| 19 | self._dataset_train_size = len(data_loader_train) |
| 20 | self._dataset_test_size = len(data_loader_test) |
| 21 | print('#train images = %d' % self._dataset_train_size) |
| 22 | print('#test images = %d' % self._dataset_test_size) |
| 23 | |
| 24 | self._model = ModelsFactory.get_by_name(self._opt.model, self._opt) |
| 25 | self._tb_visualizer = TBVisualizer(self._opt) |
| 26 | |
| 27 | self._train() |
| 28 | |
| 29 | def _train(self): |
| 30 | self._total_steps = self._opt.load_epoch * self._dataset_train_size |
nothing calls this directly
no test coverage detected