flambe.optim.scheduler

Module Contents

class flambe.optim.scheduler.LRScheduler[source]

Bases: torch.optim.lr_scheduler._LRScheduler, flambe.compile.Component

state_dict(self)[source]
class flambe.optim.scheduler.LambdaLR[source]

Bases: torch.optim.lr_scheduler.LambdaLR, flambe.compile.Component

state_dict(self)[source]