mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-22 13:30:11 +08:00
Cristobal Eyzaguirre
2924ebeda5
* moved COMET_DISABLE_AUTO_LOGGING out of modeule for flake8 compliance * Update __init__.py
25 lines
691 B
Python
25 lines
691 B
Python
from os import environ
|
|
|
|
from comet_ml import Experiment as CometExperiment
|
|
|
|
from .base import LightningLoggerBase, rank_zero_only
|
|
|
|
|
|
class CometLogger(LightningLoggerBase):
|
|
def __init__(self, *args, **kwargs):
|
|
super(CometLogger, self).__init__()
|
|
self.experiment = CometExperiment(*args, **kwargs)
|
|
|
|
@rank_zero_only
|
|
def log_hyperparams(self, params):
|
|
self.experiment.log_parameters(vars(params))
|
|
|
|
@rank_zero_only
|
|
def log_metrics(self, metrics, step_num):
|
|
# self.experiment.set_epoch(self, metrics.get('epoch', 0))
|
|
self.experiment.log_metrics(metrics)
|
|
|
|
@rank_zero_only
|
|
def finalize(self, status):
|
|
self.experiment.end()
|