Index

A

add_end_args() (in module ssfinetuning.plotting_utils) add_signature_from() (in module ssfinetuning.models)

agreement_proc() (ssfinetuning.trainer_util.TrainerForTriTraining method) append() (ssfinetuning.dataset_utils.SimpleDataset method)

B

BaseForMMTrainer (class in ssfinetuning.trainer_util)

BaseModelClass (class in ssfinetuning.models) BaseMultiPretrained (class in ssfinetuning.models)

C

check_and_replace() (in module ssfinetuning.training_args) check_for_consistency() (ssfinetuning.trainer_util.TrainerWithUWScheduler method) confi_prediction() (ssfinetuning.trainer_util.BaseForMMTrainer method) CoTrain (class in ssfinetuning.models)

cotrain() (ssfinetuning.trainer_util.TrainerForCoTraining method) cotrain_forward() (ssfinetuning.models.CoTrain method) CoTrainModelOutput (class in ssfinetuning.models) create_optimizer_and_scheduler() (ssfinetuning.trainer_util.TrainerWithUWScheduler method)

D

DefaultArgs (class in ssfinetuning.default_args)

dic_to_pandas() (in module ssfinetuning.dataset_utils) disagreement_proc() (ssfinetuning.trainer_util.TrainerForTriTraining method)

E

encode() (in module ssfinetuning.default_args) equate_lengths() (ssfinetuning.trainer_util.BaseForMMTrainer method) exchange_models() (ssfinetuning.trainer_util.TrainerForNoisyStudent method)

exchange_unlabeled_data() (ssfinetuning.trainer_util.TrainerForCoTraining method) (ssfinetuning.trainer_util.TrainerForTriTraining method) extend_length() (ssfinetuning.dataset_utils.SimpleDataset method) extract_keys() (in module ssfinetuning.dataset_utils)

F

forward() (ssfinetuning.models.CoTrain method) (ssfinetuning.models.MeanTeacher method) (ssfinetuning.models.PiModel method) (ssfinetuning.models.TemporalEnsembleModel method) (ssfinetuning.models.TriTrain method)

G

generate_kwargs() (in module ssfinetuning.training_args) get_dataloader() (ssfinetuning.trainer_util.BaseForMMTrainer method) get_default_cm() (in module ssfinetuning.default_args) get_default_legend_pos() (in module ssfinetuning.plotting_utils)

get_default_ta() (ssfinetuning.default_args.DefaultArgs method) get_default_ta_sup() (ssfinetuning.default_args.DefaultArgs method) get_linear_schedule_with_minlr() (in module ssfinetuning.trainer_util) get_train_dataloader() (ssfinetuning.trainer_util.TrainerWithUWScheduler method)

I

is_true() (ssfinetuning.trainer_util.UWScheduler method)

L

logits_m1 (ssfinetuning.models.CoTrainModelOutput attribute) (ssfinetuning.models.TriTrainModelOutput attribute) logits_m2 (ssfinetuning.models.CoTrainModelOutput attribute) (ssfinetuning.models.TriTrainModelOutput attribute)

logits_m3 (ssfinetuning.models.TriTrainModelOutput attribute) loss (ssfinetuning.models.CoTrainModelOutput attribute) (ssfinetuning.models.TriTrainModelOutput attribute)

M

m3_forward() (ssfinetuning.models.TriTrain method) match_with_batchsize() (in module ssfinetuning.dataset_utils) MeanTeacher (class in ssfinetuning.models) modify_datasets() (in module ssfinetuning.dataset_utils) module ssfinetuning ssfinetuning.dataset_utils ssfinetuning.default_args ssfinetuning.models ssfinetuning.plotting_utils ssfinetuning.trainer_util ssfinetuning.training_args

N

NoisyStudent (class in ssfinetuning.models)

num_training_steps_ (ssfinetuning.trainer_util.TrainerForNoisyStudent property)

P

PiModel (class in ssfinetuning.models) plot_in() (in module ssfinetuning.plotting_utils) plot_with_discriminator() (in module ssfinetuning.plotting_utils)

post_epoch() (ssfinetuning.trainer_util.BaseForMMTrainer method) pre_train_init() (ssfinetuning.trainer_util.BaseForMMTrainer method) prediction_step() (ssfinetuning.trainer_util.BaseForMMTrainer method) psuedo_label() (ssfinetuning.trainer_util.TrainerForNoisyStudent method)

R

reformat() (ssfinetuning.dataset_utils.SimpleDataset method)

RemoveUnusedColumnMixing (class in ssfinetuning.trainer_util) reset() (ssfinetuning.dataset_utils.SimpleDataset method)

S

set_default_args() (ssfinetuning.default_args.DefaultArgs method) set_default_vals() (in module ssfinetuning.plotting_utils) simple_forward_with_prob_logits() (ssfinetuning.models.BaseModelClass method) simple_plot() (in module ssfinetuning.plotting_utils) SimpleDataset (class in ssfinetuning.dataset_utils) sort_and_find() (in module ssfinetuning.plotting_utils) sort_and_plot() (in module ssfinetuning.plotting_utils) ssfinetuning module ssfinetuning.dataset_utils module

ssfinetuning.default_args module ssfinetuning.models module ssfinetuning.plotting_utils module ssfinetuning.trainer_util module ssfinetuning.training_args module step() (ssfinetuning.trainer_util.UWScheduler method)

T

TemporalEnsembleModel (class in ssfinetuning.models) train() (ssfinetuning.trainer_util.TrainerForCoTraining method) (ssfinetuning.trainer_util.TrainerForNoisyStudent method) (ssfinetuning.trainer_util.TrainerForTriTraining method) train_and_reset() (ssfinetuning.trainer_util.TrainerForNoisyStudent method) train_one_model() (ssfinetuning.trainer_util.TrainerForNoisyStudent method) train_with_ssl() (in module ssfinetuning.training_args) TrainerForCoTraining (class in ssfinetuning.trainer_util) TrainerForNoisyStudent (class in ssfinetuning.trainer_util) TrainerForTriTraining (class in ssfinetuning.trainer_util) TrainerWithUWScheduler (class in ssfinetuning.trainer_util)

training (ssfinetuning.models.BaseModelClass attribute) (ssfinetuning.models.BaseMultiPretrained attribute) (ssfinetuning.models.CoTrain attribute) (ssfinetuning.models.MeanTeacher attribute) (ssfinetuning.models.NoisyStudent attribute) (ssfinetuning.models.PiModel attribute) (ssfinetuning.models.TemporalEnsembleModel attribute) (ssfinetuning.models.TriTrain attribute) tri_train() (ssfinetuning.trainer_util.TrainerForTriTraining method) TriTrain (class in ssfinetuning.models) TriTrainModelOutput (class in ssfinetuning.models)

U

update_memory_logits() (ssfinetuning.models.TemporalEnsembleModel method)

update_teacher_variables() (ssfinetuning.models.MeanTeacher method) UWScheduler (class in ssfinetuning.trainer_util)

W

with_labeled_fraction() (in module ssfinetuning.training_args)

wrapper_for_l_fr() (in module ssfinetuning.training_args)

Z

zero_teacher_weights() (ssfinetuning.models.MeanTeacher method)