mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-22 13:30:11 +08:00
Jirka Borovec
37647d835a
* add package info #358 * Update __init__.py * wrap package info * update CI * fix package info * fix for #388 * prune duplicated configs * fix install * use req from file * move info to sep. module drop comments from req * add setup req. * add setup req. * update get info * refactor init * update pip * fix failing on buildins * fix failing open * fix test imports * fix tests * fix pep8
26 lines
747 B
Python
26 lines
747 B
Python
try:
|
|
from comet_ml import Experiment as CometExperiment
|
|
except ImportError:
|
|
raise ImportError('Missing comet_ml package.')
|
|
|
|
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()
|