def _pre_experiment_hook(self, experiment: Experiment):
monitoring_params = experiment.monitoring_params
monitoring_params["dir"] = str(Path(experiment.logdir).absolute())
neptune.init(**monitoring_params["init"])
self._neptune_experiment = neptune.create_experiment(
**monitoring_params["create_experiment"])
log_on_batch_end: bool = monitoring_params.pop("log_on_batch_end",
False)
log_on_epoch_end: bool = monitoring_params.pop("log_on_epoch_end",
True)
self._init(
log_on_batch_end=log_on_batch_end,
log_on_epoch_end=log_on_epoch_end,
)
self._neptune_experiment.set_property("log_on_batch_end",
self.log_on_batch_end)
self._neptune_experiment.set_property("log_on_epoch_end",
self.log_on_epoch_end)
if isinstance(experiment, ConfigExperiment):
exp_config = utils.flatten_dict(experiment.stages_config)
for name, value in exp_config.items():
self._neptune_experiment.set_property(name, value)
def _pre_experiment_hook(self, experiment: Experiment):
monitoring_params = experiment.monitoring_params
monitoring_params["dir"] = str(Path(experiment.logdir).absolute())
log_on_batch_end: bool = \
monitoring_params.pop("log_on_batch_end", False)
log_on_epoch_end: bool = \
monitoring_params.pop("log_on_epoch_end", True)
checkpoints_glob: List[str] = \
monitoring_params.pop("checkpoints_glob", None)
self._init(
log_on_batch_end=log_on_batch_end,
log_on_epoch_end=log_on_epoch_end,
checkpoints_glob=checkpoints_glob,
)
if isinstance(experiment, ConfigExperiment):
exp_config = utils.flatten_dict(experiment.stages_config)
wandb.init(**monitoring_params, config=exp_config)
else:
wandb.init(**monitoring_params)
