tuners.loggers package

Submodules

tuners.loggers.base module

class tuners.loggers.base.BaseTunerObserver(tuner)[source]

Bases: object

on_eval_samples_updated(context, samples, proposal_log_scores, model_log_scores, target_log_scores)[source]
on_metric_updated(name, value)[source]
on_ministep_idx_updated(s)[source]
on_parameters_updated(params)[source]
on_step_idx_updated(s)[source]

tuners.loggers.console module

class tuners.loggers.console.ConsoleLogger(tuner)[source]

Bases: BaseTunerObserver

on_ministep_idx_updated(s)[source]
on_parameters_updated(params)[source]
on_step_idx_updated(s)[source]

tuners.loggers.json module

class tuners.loggers.json.JSONLogger(tuner, project, name, path='/gfs-ssd/user/gkruszew/breaking-bad/disco', save_steps=1, store_eval_samples=False, **kwargs)[source]

Bases: BaseTunerObserver

Reports DPGTuner statistics to a JSON file

on_eval_samples_updated(context, samples, proposal_log_scores, model_log_scores, target_log_scores)[source]
on_metric_updated(name, value)[source]
on_ministep_idx_updated(s)[source]
on_parameters_updated(params)[source]
on_step_idx_updated(s)[source]
save()[source]

tuners.loggers.neptune module

class tuners.loggers.neptune.NeptuneLogger(tuner, project, name=None, api_token=None, **kwargs)[source]

Bases: BaseTunerObserver

Reports DPGTuner statistics to Neptune

on_eval_samples_updated(context, samples, proposal_log_scores, model_log_scores, target_log_scores)[source]
on_metric_updated(name, value)[source]
on_ministep_idx_updated(s)[source]
on_parameters_updated(params)[source]
on_step_idx_updated(s)[source]
tuners.loggers.neptune.get_proxies()[source]

tuners.loggers.wandb module

class tuners.loggers.wandb.WandBLogger(tuner, project, name=None)[source]

Bases: BaseTunerObserver

Reports DPGTuner statistics to Weights & Biases

on_eval_samples_updated(context, samples, proposal_log_scores, model_log_scores, target_log_scores)[source]
on_metric_updated(name, value)[source]
on_ministep_idx_updated(m)[source]
on_parameters_updated(params)[source]
on_step_idx_updated(s)[source]

Module contents