Skip to content
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

new feature for profiling training runs #782

Merged
merged 24 commits into from
Feb 7, 2020
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
allow profiler=True creation
  • Loading branch information
Jeremy Jordan committed Feb 6, 2020
commit 474785f6e488c9ca40e841c18553357413ea9bda
4 changes: 3 additions & 1 deletion pytorch_lightning/trainer/trainer.py
Original file line number Diff line number Diff line change
@@ -26,7 +26,7 @@
from pytorch_lightning.trainer.training_loop import TrainerTrainLoopMixin
from pytorch_lightning.trainer.training_tricks import TrainerTrainingTricksMixin
from pytorch_lightning.utilities.debugging import MisconfigurationException
from pytorch_lightning.utilities.profiler import PassThroughProfiler
from pytorch_lightning.utilities.profiler import Profiler, PassThroughProfiler


try:
@@ -578,6 +578,8 @@ def __init__(
self.configure_logger(logger)

# configure profiler
if profiler is True:
profiler = Profiler()
self.profiler = profiler or PassThroughProfiler()

# configure early stop callback