d3rlpy.logging.WanDBAdapter

class d3rlpy.logging.WanDBAdapter(experiment_name, project=None)

WandB Logger Adapter class.

This class logs data to Weights & Biases (WandB) for experiment tracking.

Parameters

experiment_name (str) – Name of the experiment.

Methods

after_write_metric(epoch, step)

Callback executed after writing metric.

Parameters

• epoch (int) –

• step (int) –

Return type

None

before_write_metric(epoch, step)

Callback executed before writing metric.

Parameters

• epoch (int) –

• step (int) –

Return type

None

close()

Closes the logger and finishes the WandB run.

Return type

None

save_model(epoch, algo)

Saves models to Weights & Biases.

Not implemented for WandB.

Parameters

• epoch (int) –

• algo (d3rlpy.logging.logger.SaveProtocol) –

Return type

None

write_metric(epoch, step, name, value)

Writes metric to WandB.

Parameters

• epoch (int) –

• step (int) –

• name (str) –

• value (float) –

Return type

None

write_params(params)

Writes hyperparameters to WandB config.

Parameters

params (Dict[str, Any]) –

Return type

None