nntoolbox.callbacks.transfer module

class nntoolbox.callbacks.transfer.FineTuning(unfreeze_every: int, freeze_inds: Optional[List[int]] = None, lr: Optional[Union[List[float], float]] = None)

Bases: nntoolbox.callbacks.callbacks.GroupCallback

Combining freezing batch norm and gradual unfreezing of layer

class nntoolbox.callbacks.transfer.FreezeBN

Bases: nntoolbox.callbacks.callbacks.Callback

Freeze statistics of non trainable batch norms so that it won’t accumulate statistics (UNTESTED)

on_epoch_begin()

class nntoolbox.callbacks.transfer.GradualUnfreezing(unfreeze_every: int, freeze_inds: Optional[List[int]] = None, lr: Optional[Union[List[float], float]] = None)

Bases: nntoolbox.callbacks.callbacks.Callback

Gradually unfreezing pretrained layers, with discriminative learning rates (UNTESTED)

on_epoch_end(logs: Dict[str, Any]) → bool

on_train_begin()