-
Notifications
You must be signed in to change notification settings - Fork 3.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Remove LoggerConnector.on_trainer_init
#11121
Remove LoggerConnector.on_trainer_init
#11121
Conversation
- Moves property initialization to trainer - Adds typehints for affected flags on the Trainer
@@ -48,25 +46,6 @@ def __init__(self, trainer: "pl.Trainer", log_gpu_memory: Optional[str] = None) | |||
self._batch_idx: Optional[int] = None | |||
self._split_idx: Optional[int] = None | |||
|
|||
def on_trainer_init( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
In #10417 we propose to make the logger owned by the connector. The two methods on_trainer_init
and configure_logger
would then essentially become part of the LoggerConnector init without reference to the trainer. What are your thoughts?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I would definitely prefer this approach. Right now, the Trainer is starting to become a bigger class and it is getting harder for users to parse it.
@@ -569,9 +569,9 @@ def __init__( | |||
# configure profiler | |||
self.__init_profiler(profiler) | |||
|
|||
# init logger flags | |||
# configure logger flags | |||
self.logger: Optional[LightningLoggerBase] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I played a bit with this and this definition needs to be in the __init__
, otherwise mypy will treat it as Any
What does this PR do?
Follow up to comment here: https://github.com/PyTorchLightning/pytorch-lightning/pull/11114/files#r771039150
Part of #7037, #10417
Does your PR introduce any breaking changes? If yes, please list them.
None
Before submitting
PR review
Anyone in the community is welcome to review the PR.
Before you start reviewing make sure you have read Review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃