KLDivergenceBCELoss.ConfigΒΆ

Component: KLDivergenceBCELoss

class KLDivergenceBCELoss.Config[source]

Bases: ConfigBase

All Attributes (including base classes)

temperature: float = 1.0
hard_weight: float = 0.0

Default JSON

{
    "temperature": 1.0,
    "hard_weight": 0.0
}