fovi.training package

Experiment training, datasets, and diagnostics (fovi[training]).

Subpackages

• fovi.training.utils package
  • fovi.training.utils.backup
    • backup_model()
    • backup_models()
  • fovi.training.utils.flops
    • make_flop_counter()
    • FlopWrapper
      • FlopWrapper.trainer
      • FlopWrapper.kwargs
      • FlopWrapper.__init__()
      • FlopWrapper.get_inputs()
      • FlopWrapper.forward()
    • measure_latency()
    • get_flops_df()
  • fovi.training.utils.knnprobe
    • knn_probe()
    • knn_classifier()
  • fovi.training.utils.losses
    • SimCLRLoss
      • SimCLRLoss.__init__()
      • SimCLRLoss.mask_correlated_samples()
      • SimCLRLoss.compute_logits()
      • SimCLRLoss.forward()
      • SimCLRLoss.accuracy()
    • VicRegLoss
      • VicRegLoss.sim_coeff
      • VicRegLoss.std_coeff
      • VicRegLoss.cov_coeff
      • VicRegLoss.__init__()
      • VicRegLoss.forward()
    • BarlowTwinsLoss
      • BarlowTwinsLoss.bn
      • BarlowTwinsLoss.lambd
      • BarlowTwinsLoss.batch_size
      • BarlowTwinsLoss.world_size
      • BarlowTwinsLoss.__init__()
      • BarlowTwinsLoss.forward()
  • fovi.training.utils.lr_scheduling
    • LARS
      • LARS.__init__()
      • LARS.step()
    • trunc_normal_()
    • cosine_decay()
    • cosine_scheduler()
    • learning_schedule()
    • CosineDecayWithWarmup
      • CosineDecayWithWarmup.scaled_base_lr
      • CosineDecayWithWarmup.end_lr
      • CosineDecayWithWarmup.__init__()
      • CosineDecayWithWarmup.state_dict()
      • CosineDecayWithWarmup.load_state_dict()

Modules

• fovi.training.loader
  • FlashLoader
    • FlashLoader.__init__()
• fovi.training.loading
  • Sequence
    • Sequence.count()
    • Sequence.index()
  • OmegaConf
    • OmegaConf.__init__()
    • OmegaConf.structured()
    • OmegaConf.create()
    • OmegaConf.load()
    • OmegaConf.save()
    • OmegaConf.from_cli()
    • OmegaConf.from_dotlist()
    • OmegaConf.merge()
    • OmegaConf.unsafe_merge()
    • OmegaConf.register_resolver()
    • OmegaConf.legacy_register_resolver()
    • OmegaConf.register_new_resolver()
    • OmegaConf.has_resolver()
    • OmegaConf.clear_resolvers()
    • OmegaConf.clear_resolver()
    • OmegaConf.get_cache()
    • OmegaConf.set_cache()
    • OmegaConf.clear_cache()
    • OmegaConf.copy_cache()
    • OmegaConf.set_readonly()
    • OmegaConf.is_readonly()
    • OmegaConf.set_struct()
    • OmegaConf.is_struct()
    • OmegaConf.masked_copy()
    • OmegaConf.to_container()
    • OmegaConf.to_object()
    • OmegaConf.is_missing()
    • OmegaConf.is_interpolation()
    • OmegaConf.is_list()
    • OmegaConf.is_dict()
    • OmegaConf.is_config()
    • OmegaConf.get_type()
    • OmegaConf.select()
    • OmegaConf.update()
    • OmegaConf.to_yaml()
    • OmegaConf.resolve()
    • OmegaConf.missing_keys()
  • open_dict()
  • find_config()
  • HiddenPrints
    • HiddenPrints.__init__()
  • Trainer
    • Trainer.__init__()
    • Trainer.setup_distributed()
    • Trainer.cleanup_distributed()
    • Trainer.create_optimizer()
    • Trainer.create_train_loader()
    • Trainer.create_val_loader()
    • Trainer.create_standard_loader()
    • Trainer.create_model_and_scaler()
    • Trainer.reset_model()
    • Trainer.train()
    • Trainer.eval_and_log()
    • Trainer.load_checkpoint()
    • Trainer.checkpoint()
    • Trainer.save_checkpoint()
    • Trainer.train_loop()
    • Trainer.val_loop()
    • Trainer.compute_activations()
    • Trainer.initialize_logger()
    • Trainer.copy_hydra_outputs()
    • Trainer.initialize_remote_logger()
    • Trainer.log()
    • Trainer.exec()
    • Trainer.launch_from_args()
    • Trainer.add_supervised_meters()
    • Trainer.final_accuracy()
  • get_trainer_from_base_fn()
• fovi.training.trainer
  • Trainer
    • Trainer.__init__()
    • Trainer.setup_distributed()
    • Trainer.cleanup_distributed()
    • Trainer.create_optimizer()
    • Trainer.create_train_loader()
    • Trainer.create_val_loader()
    • Trainer.create_standard_loader()
    • Trainer.create_model_and_scaler()
    • Trainer.reset_model()
    • Trainer.train()
    • Trainer.eval_and_log()
    • Trainer.load_checkpoint()
    • Trainer.checkpoint()
    • Trainer.save_checkpoint()
    • Trainer.train_loop()
    • Trainer.val_loop()
    • Trainer.compute_activations()
    • Trainer.initialize_logger()
    • Trainer.copy_hydra_outputs()
    • Trainer.initialize_remote_logger()
    • Trainer.log()
    • Trainer.exec()
    • Trainer.launch_from_args()
    • Trainer.add_supervised_meters()
    • Trainer.final_accuracy()
• fovi.training.visualizer
  • Visualizer
    • Visualizer.trainer
    • Visualizer.model
    • Visualizer.out_base_dir
    • Visualizer.out_dir
    • Visualizer.__init__()
    • Visualizer.multi_fixation_accuracy()
    • Visualizer.visualize_filters()
    • Visualizer.visualize_filter()
    • Visualizer.visualize_filter_over_space()
    • Visualizer.plot_coordinates()
    • Visualizer.plot_spatial_rf_from_multi_layers()
    • Visualizer.plot_sampling_grids()
    • Visualizer.plot_rf_diameters()
    • Visualizer.print_all_functions()
  • get_first_conv_weights()
  • visualize_filters()
  • visualize_filters_knn()