Probe

class core.finetuning.Probe(model, cfg, unfrozen_prefixes)

Bases: core.finetuning.base.FinetuningStrategy

Freeze all parameters except those matching unfrozen_prefixes.

Parameters:

unfrozen_prefixes (list[str]) – parameter name prefixes to keep trainable