Skip to content

Commit 1d4b6be

Browse files
BordawilliamFalcon
authored andcommitted
rename trainer modules, drop _mixin (Lightning-AI#571)
* rename trainer modules, drop _mixin * fix imports
1 parent e0dbc8a commit 1d4b6be

File tree

13 files changed

+14
-14
lines changed

13 files changed

+14
-14
lines changed

pytorch_lightning/trainer/amp_mixin.py renamed to pytorch_lightning/trainer/auto_mix_precision.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def init_amp(self, use_amp):
1818

1919
if use_amp and not APEX_AVAILABLE: # pragma: no cover
2020
msg = """
21-
You set use_amp=True but do not have apex installed.
21+
You set `use_amp=True` but do not have apex installed.
2222
Install apex first using this guide and rerun with use_amp=True:
2323
https://github.com/NVIDIA/apex#linux
2424
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

pytorch_lightning/trainer/trainer.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -13,21 +13,21 @@
1313
import tqdm
1414
from torch.optim.optimizer import Optimizer
1515

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,
2222
parse_gpu_ids,
2323
determine_root_gpu_device
2424
)
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
2929
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
3131
from pytorch_lightning.utilities.debugging import MisconfigurationException
3232

3333
try:
File renamed without changes.

0 commit comments

Comments
 (0)