FinetuningStrategy#

class core.finetuning.FinetuningStrategy(model, cfg)[source]#

Bases: abc.ABC

Base class for finetuning strategies.

Subclasses must implement:

  • setup: initialize frozen/unfrozen parameter state

  • update: called at the start of each epoch to adjust parameter state