cascade.nn.Module

class cascade.nn.Module(*args, **kwargs)[source]

Bases: Module

Abstract module class supporting parameter freezing, decayed / non-decayed parameter iteration, and cached property clearing

Methods

clear_cached_properties

Clear cached properties to allow re-calculation

decay_params

Iterate through weight decayed parameters

freeze_params

Freeze parameters and turn on evaluation mode

regular_params

Iterate through non-decayed parameters

unfreeze_params

Unfreeze parameters and turn on training mode

Attributes

frozen