Skip to content

Commit

Permalink
drop logging level (#1015)
Browse files Browse the repository at this point in the history
  • Loading branch information
Borda committed Mar 3, 2020
1 parent 2a04be0 commit 04c9eb4
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 8 deletions.
3 changes: 0 additions & 3 deletions pytorch_lightning/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,6 @@
# We are not importing the rest of the scikit during the build
# process, as it may not be compiled yet
else:
import logging as log
log.basicConfig(level=log.INFO)

from .core import data_loader, LightningModule
from .trainer import Trainer
from .callbacks import Callback
Expand Down
8 changes: 3 additions & 5 deletions pytorch_lightning/profiler/profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,7 @@
import pstats
import io
from abc import ABC, abstractmethod
import logging

logger = logging.getLogger(__name__)
import logging as log


class BaseProfiler(ABC):
Expand Down Expand Up @@ -124,7 +122,7 @@ def log_row(action, mean, total):
action, f"{np.mean(durations):.5}", f"{np.sum(durations):.5}",
)
output_string += "\n"
logger.info(output_string)
log.info(output_string)


class AdvancedProfiler(BaseProfiler):
Expand Down Expand Up @@ -177,4 +175,4 @@ def describe(self):
output_string = "\nProfiler Report\n"
for action, stats in self.recorded_stats.items():
output_string += f"\nProfile stats for: {action}\n{stats}"
logger.info(output_string)
log.info(output_string)

0 comments on commit 04c9eb4

Please sign in to comment.