torchref.refinement.targets.adp.entropy module

class torchref.refinement.targets.adp.entropy.ADPEntropyTarget(model=None, verbose=0)

Bases: ADPTarget

ADP Entropy regularization target.

Uses the model’s existing adp_kl_divergence_loss or similar.

name: str = 'adp/KL'

__init__(model=None, verbose=0)

Initialize model target.

Parameters:

• model (Model, optional) – Reference to the Model object (optional for empty init).

• verbose (int, optional) – Verbosity level. Default is 0.

forward()

Compute and return the loss. Override in subclasses.

stats()

Get KL divergence statistics.