Index A | B | C | E | F | G | H | L | M | P | R | S | T | U | V A attach_hooks() (visualizing_training.train.ModelManager method) Attention (class in visualizing_training.model) B break_list_by_lengths() (in module visualizing_training.utils) C characterize_all_transitions() (in module visualizing_training.utils) characterize_transition_between_phases() (in module visualizing_training.utils) clear_metrics_cache() (visualizing_training.train.ModelManager method) clear_weights_biases_cache() (visualizing_training.train.ModelManager method) compute_metrics() (visualizing_training.train.ModelManager method) conf module E Embed (class in visualizing_training.model) F feature_importance() (visualizing_training.hmm.HMM method) FF (class in visualizing_training.model) find_i_followed_by_j() (in module visualizing_training.utils) forward() (visualizing_training.model.Attention method) (visualizing_training.model.Embed method) (visualizing_training.model.FF method) (visualizing_training.model.HookPoint method) (visualizing_training.model.LeNet5 method) (visualizing_training.model.MLP method) (visualizing_training.model.MyBasicBlock method) (visualizing_training.model.PosEmbed method) (visualizing_training.model.ToyModel method) (visualizing_training.model.Transformer method) (visualizing_training.model.TransformerBlock method) (visualizing_training.model.Unembed method) G get_avg_log_likelihood() (visualizing_training.hmm.HMM method) get_derivatives() (in module visualizing_training.utils) get_difference_bt_means() (in module visualizing_training.utils) get_features_for_transition() (in module visualizing_training.utils) H HMM (class in visualizing_training.hmm) HookPoint (class in visualizing_training.model) L LeNet5 (class in visualizing_training.model) load_model() (in module visualizing_training.utils) M make_hmm_data() (in module visualizing_training.utils) MLP (class in visualizing_training.model) ModelManager (class in visualizing_training.train) module conf visualizing_training.hmm visualizing_training.model visualizing_training.train visualizing_training.utils visualizing_training.visualize MyBasicBlock (class in visualizing_training.model) MyResNet (class in visualizing_training.model) P PosEmbed (class in visualizing_training.model) R remove_hooks() (visualizing_training.train.ModelManager method) S save_metrics() (visualizing_training.train.ModelManager method) save_model() (in module visualizing_training.utils) softmax_with_overflow() (in module visualizing_training.utils) T ToyModel (class in visualizing_training.model) train() (visualizing_training.train.ModelManager method) train_and_save_metrics() (visualizing_training.train.ModelManager method) Transformer (class in visualizing_training.model) TransformerBlock (class in visualizing_training.model) U Unembed (class in visualizing_training.model) V visualize_all_seeds() (in module visualizing_training.visualize) visualize_avg_log_likelihood() (in module visualizing_training.visualize) visualize_hmm_loss() (in module visualizing_training.visualize) visualize_states() (in module visualizing_training.visualize) visualizing_training.hmm module visualizing_training.model module visualizing_training.train module visualizing_training.utils module visualizing_training.visualize module