Closed ptigwe closed 2 years ago
This adds a trainable_parameters filter which filters out parts of a given tx.Module which have been frozen. Also modified the docs to reflect this change and how it can be used when training a model with frozen sections.
trainable_parameters
tx.Module
This adds a
trainable_parameters
filter which filters out parts of a giventx.Module
which have been frozen. Also modified the docs to reflect this change and how it can be used when training a model with frozen sections.