|
13 | 13 | import tqdm |
14 | 14 | from torch.optim.optimizer import Optimizer |
15 | 15 |
|
16 | | -from pytorch_lightning.trainer.amp_mixin import TrainerAMPMixin |
17 | | -from pytorch_lightning.trainer.callback_config_mixin import TrainerCallbackConfigMixin |
18 | | -from pytorch_lightning.trainer.data_loading_mixin import TrainerDataLoadingMixin |
19 | | -from pytorch_lightning.trainer.ddp_mixin import TrainerDDPMixin |
20 | | -from pytorch_lightning.trainer.dp_mixin import TrainerDPMixin |
21 | | -from pytorch_lightning.trainer.dp_mixin import ( |
| 16 | +from pytorch_lightning.trainer.auto_mix_precision import TrainerAMPMixin |
| 17 | +from pytorch_lightning.trainer.callback_config import TrainerCallbackConfigMixin |
| 18 | +from pytorch_lightning.trainer.data_loading import TrainerDataLoadingMixin |
| 19 | +from pytorch_lightning.trainer.distrib_data_parallel import TrainerDDPMixin |
| 20 | +from pytorch_lightning.trainer.distrib_parts import ( |
| 21 | + TrainerDPMixin, |
22 | 22 | parse_gpu_ids, |
23 | 23 | determine_root_gpu_device |
24 | 24 | ) |
25 | | -from pytorch_lightning.trainer.evaluation_loop_mixin import TrainerEvaluationLoopMixin |
26 | | -from pytorch_lightning.trainer.logging_mixin import TrainerLoggingMixin |
27 | | -from pytorch_lightning.trainer.model_hooks_mixin import TrainerModelHooksMixin |
28 | | -from pytorch_lightning.trainer.train_loop_mixin import TrainerTrainLoopMixin |
| 25 | +from pytorch_lightning.trainer.evaluation_loop import TrainerEvaluationLoopMixin |
| 26 | +from pytorch_lightning.trainer.logging import TrainerLoggingMixin |
| 27 | +from pytorch_lightning.trainer.model_hooks import TrainerModelHooksMixin |
| 28 | +from pytorch_lightning.trainer.training_loop import TrainerTrainLoopMixin |
29 | 29 | from pytorch_lightning.trainer.trainer_io import TrainerIOMixin |
30 | | -from pytorch_lightning.trainer.training_tricks_mixin import TrainerTrainingTricksMixin |
| 30 | +from pytorch_lightning.trainer.training_tricks import TrainerTrainingTricksMixin |
31 | 31 | from pytorch_lightning.utilities.debugging import MisconfigurationException |
32 | 32 |
|
33 | 33 | try: |
|
0 commit comments