tiny_diff.losses.vae_loss.BetaPolicy

class tiny_diff.losses.vae_loss.BetaPolicy(a: float = 0.001, b: float = 0.01, offset: float = 60, slope: float = 5)

Bases: Callable

Function representing a Beta Policy.

This sigmoid like function is used to reflect a weight that changes over each epoch.

Parameters:

• a – minimum value

• b – maximum value

• offset – location of sigmoid’s middle

• slope – rate of growth. the higher the sharper is the growth.

__init__(a: float = 0.001, b: float = 0.01, offset: float = 60, slope: float = 5)

Methods

__init__([a, b, offset, slope])