Index

Symbols

(class (finetuning_scheduler.fts_supporters.ScheduleParsingMixin.reinit_lr_scheduler parameter)

_

_fts_state (finetuning_scheduler.fts.FinetuningScheduler attribute)

A

add_optimizer_groups() (finetuning_scheduler.fts_supporters.ScheduleImplMixin static method) agg_func (finetuning_scheduler.fts_supporters.ScheduleImplMixin.sync parameter) allow_untested (finetuning_scheduler.fts.FinetuningScheduler parameter), [1]

apply_lambdas (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter) apply_lambdas_new_pgs (finetuning_scheduler.fts.FinetuningScheduler parameter), [1] asets (finetuning_scheduler.fts_supporters.ScheduleImplMixin.sync parameter)

B

base_max_lr (finetuning_scheduler.fts.FinetuningScheduler parameter), [1]

best_ckpt_depth (finetuning_scheduler.fts_supporters.FTSCheckpoint attribute)

C

callback_attrs (finetuning_scheduler.fts_supporters.CallbackResolverMixin parameter), [1] callback_parents (finetuning_scheduler.fts_supporters.CallbackResolverMixin parameter), [1] CallbackDepMixin (class in finetuning_scheduler.fts_supporters) CallbackResolverMixin (class in finetuning_scheduler.fts_supporters)

check_on_train_epoch_end (finetuning_scheduler.fts_supporters.FTSEarlyStopping attribute) connect_callback() (finetuning_scheduler.fts_supporters.CallbackResolverMixin method) construct_mapping() (finetuning_scheduler.fts_supporters.UniqueKeyLoader method) curr_depth (finetuning_scheduler.fts.FinetuningScheduler property) current_ckpt_depth (finetuning_scheduler.fts_supporters.FTSCheckpoint attribute)

D

depth (finetuning_scheduler.fts.FinetuningScheduler.step_pg parameter) (finetuning_scheduler.fts_supporters.ScheduleImplMixin.thaw_to_depth parameter) depth_remaining (finetuning_scheduler.fts.FinetuningScheduler property)

depth_sync (finetuning_scheduler.fts.FinetuningScheduler.step_pg parameter) dump_loc (finetuning_scheduler.fts_supporters.ScheduleImplMixin.gen_ft_schedule parameter) (finetuning_scheduler.fts_supporters.ScheduleImplMixin.save_schedule parameter)

E

epoch_transitions_only (finetuning_scheduler.fts.FinetuningScheduler parameter), [1]

es_phase_complete (finetuning_scheduler.fts_supporters.FTSEarlyStopping attribute) exec_ft_phase() (finetuning_scheduler.fts_supporters.ScheduleImplMixin static method)

F

final_phase (finetuning_scheduler.fts_supporters.FTSEarlyStopping attribute) finetuning_scheduler.fts module finetuning_scheduler.fts_supporters module FinetuningScheduler (class in finetuning_scheduler.fts)

finetuningscheduler_callback (finetuning_scheduler.fts_supporters.FTSCheckpoint attribute) (finetuning_scheduler.fts_supporters.FTSEarlyStopping attribute) freeze_before_training() (finetuning_scheduler.fts.FinetuningScheduler method) ft_schedule (finetuning_scheduler.fts.FinetuningScheduler parameter), [1] FTSCheckpoint (class in finetuning_scheduler.fts_supporters) FTSEarlyStopping (class in finetuning_scheduler.fts_supporters) FTSState (class in finetuning_scheduler.fts_supporters)

G

gen_ft_sched_only (finetuning_scheduler.fts.FinetuningScheduler parameter), [1] gen_ft_schedule() (finetuning_scheduler.fts_supporters.ScheduleImplMixin static method)

gen_implicit_schedule() (finetuning_scheduler.fts_supporters.ScheduleImplMixin method) gen_or_load_sched() (finetuning_scheduler.fts_supporters.ScheduleImplMixin method)

I

init_ft_sched() (finetuning_scheduler.fts_supporters.ScheduleImplMixin method) init_fts() (finetuning_scheduler.fts_supporters.ScheduleImplMixin method)

init_thaw (finetuning_scheduler.fts_supporters.ScheduleImplMixin.exec_ft_phase parameter) initial_denom_lr (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter)

L

layer_config (finetuning_scheduler.fts_supporters.ScheduleImplMixin.save_schedule parameter) load_state_dict() (finetuning_scheduler.fts.FinetuningScheduler method) (finetuning_scheduler.fts_supporters.FTSCheckpoint method)

load_yaml_schedule() (finetuning_scheduler.fts_supporters.ScheduleImplMixin static method) lr (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter)

M

max_depth (finetuning_scheduler.fts.FinetuningScheduler parameter), [1] module finetuning_scheduler.fts finetuning_scheduler.fts_supporters

module (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter) (finetuning_scheduler.fts_supporters.ScheduleImplMixin.exec_ft_phase parameter) (finetuning_scheduler.fts_supporters.ScheduleImplMixin.gen_ft_schedule parameter)

N

new_lr_scheduler (finetuning_scheduler.fts_supporters.ScheduleParsingMixin.reinit_lr_scheduler parameter)

no_decay (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter)

O

objs (finetuning_scheduler.fts_supporters.ScheduleImplMixin.sync parameter) on_before_zero_grad() (finetuning_scheduler.fts.FinetuningScheduler method) on_fit_start() (finetuning_scheduler.fts.FinetuningScheduler method) on_train_end() (finetuning_scheduler.fts.FinetuningScheduler method)

on_train_epoch_start() (finetuning_scheduler.fts.FinetuningScheduler method) optimizer (finetuning_scheduler.fts.FinetuningScheduler.on_before_zero_grad parameter) (finetuning_scheduler.fts.FinetuningScheduler.step_pg parameter) (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter)

P

pl_module (finetuning_scheduler.fts.FinetuningScheduler.freeze_before_training parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_before_zero_grad parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_fit_start parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_train_end parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_train_epoch_start parameter) (finetuning_scheduler.fts.FinetuningScheduler.setup parameter)

R

reconnect (finetuning_scheduler.fts_supporters.CallbackResolverMixin.connect_callback parameter) reinit_lr_cfg (finetuning_scheduler.fts.FinetuningScheduler parameter), [1]

reinit_lr_scheduler() (finetuning_scheduler.fts_supporters.ScheduleParsingMixin method) restore_best (finetuning_scheduler.fts.FinetuningScheduler parameter), [1] restore_best_ckpt() (finetuning_scheduler.fts.FinetuningScheduler method)

S

save_on_train_epoch_end (finetuning_scheduler.fts_supporters.FTSCheckpoint attribute) save_schedule() (finetuning_scheduler.fts_supporters.ScheduleImplMixin static method) sched_dir (finetuning_scheduler.fts_supporters.ScheduleImplMixin.gen_implicit_schedule parameter) schedule_name (finetuning_scheduler.fts_supporters.ScheduleImplMixin.save_schedule parameter) schedule_yaml_file (finetuning_scheduler.fts_supporters.ScheduleImplMixin.load_yaml_schedule parameter) ScheduleImplMixin (class in finetuning_scheduler.fts_supporters) ScheduleParsingMixin (class in finetuning_scheduler.fts_supporters) setup() (finetuning_scheduler.fts.FinetuningScheduler method) (finetuning_scheduler.fts_supporters.FTSCheckpoint method) (finetuning_scheduler.fts_supporters.FTSEarlyStopping method)

should_transition() (finetuning_scheduler.fts.FinetuningScheduler method) stage (finetuning_scheduler.fts.FinetuningScheduler.setup parameter) state_dict (finetuning_scheduler.fts.FinetuningScheduler.load_state_dict parameter) (finetuning_scheduler.fts_supporters.FTSCheckpoint.load_state_dict parameter) state_dict() (finetuning_scheduler.fts.FinetuningScheduler method) (finetuning_scheduler.fts_supporters.FTSCheckpoint method) step() (finetuning_scheduler.fts.FinetuningScheduler method) step_pg() (finetuning_scheduler.fts.FinetuningScheduler method) support_multiple (finetuning_scheduler.fts_supporters.CallbackResolverMixin parameter), [1] sync() (finetuning_scheduler.fts_supporters.ScheduleImplMixin static method)

T

target_callback_ref (finetuning_scheduler.fts_supporters.CallbackResolverMixin parameter), [1] thaw_pl (finetuning_scheduler.fts_supporters.ScheduleImplMixin.exec_ft_phase parameter) thaw_to_depth() (finetuning_scheduler.fts_supporters.ScheduleImplMixin method) thawed_pl (finetuning_scheduler.fts_supporters.ScheduleImplMixin.add_optimizer_groups parameter) trainer (finetuning_scheduler.fts.FinetuningScheduler.on_before_zero_grad parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_fit_start parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_train_end parameter) (finetuning_scheduler.fts.FinetuningScheduler.on_train_epoch_start parameter) (finetuning_scheduler.fts.FinetuningScheduler.setup parameter) (finetuning_scheduler.fts.FinetuningScheduler.should_transition parameter) (finetuning_scheduler.fts_supporters.CallbackResolverMixin.connect_callback parameter) (finetuning_scheduler.fts_supporters.FTSCheckpoint.setup parameter) (finetuning_scheduler.fts_supporters.ScheduleParsingMixin.reinit_lr_scheduler parameter)

U

UniqueKeyLoader (class in finetuning_scheduler.fts_supporters)