elektronn3.training package

Submodules

• elektronn3.training.handlers module
  • get_cmap()
  • plot_image()
  • write_to_kzip()
• elektronn3.training.metrics module
  • AMI
    • AMI.name
  • ARI
    • ARI.name
  • AUROC
    • AUROC.name
  • Accuracy
    • Accuracy.name
  • AveragePrecision
    • AveragePrecision.name
  • DSC
    • DSC.name
  • Evaluator
    • Evaluator.name
  • IoU
    • IoU.name
  • NMI
    • NMI.name
  • Precision
    • Precision.name
  • Recall
    • Recall.name
  • SilhouetteScore
    • SilhouetteScore.name
  • accuracy()
  • auroc()
  • average_precision()
  • bin_accuracy()
  • bin_auroc()
  • bin_average_precision()
  • bin_dice_coefficient()
  • bin_iou()
  • bin_precision()
  • bin_recall()
  • channel_metric()
  • confusion_matrix()
  • dice_coefficient()
  • iou()
  • precision()
  • recall()
• elektronn3.training.noise2void module
  • Noise2VoidTrainer
    • Noise2VoidTrainer.epoch
    • Noise2VoidTrainer.exp_name
    • Noise2VoidTrainer.out_channels
    • Noise2VoidTrainer.save_path
    • Noise2VoidTrainer.step
    • Noise2VoidTrainer.tb
    • Noise2VoidTrainer.terminate
    • Noise2VoidTrainer.train_loader
    • Noise2VoidTrainer.valid_loader
  • get_stratified_coords()
  • prepare_sample()
• elektronn3.training.padam module
  • Padam
    • Padam.step()
• elektronn3.training.plotting module
  • Scroller
    • Scroller.onscroll()
    • Scroller.update()
  • add_timeticks()
  • plot_hist()
  • scroll_plot()
  • sma()
• elektronn3.training.recalibration module
  • NoApplicableLayersException
  • recalibrate_bn()
• elektronn3.training.swa module
  • SWA
    • SWA.add_param_group()
    • SWA.bn_update()
    • SWA.load_state_dict()
    • SWA.state_dict()
    • SWA.step()
    • SWA.swap_swa_sgd()
    • SWA.update_swa()
    • SWA.update_swa_group()
• elektronn3.training.train_utils module
  • AccumulationArray
    • AccumulationArray.add_offset()
    • AccumulationArray.append()
    • AccumulationArray.clear()
    • AccumulationArray.data
    • AccumulationArray.ema
    • AccumulationArray.max()
    • AccumulationArray.mean()
    • AccumulationArray.min()
    • AccumulationArray.sum()
  • HistoryTracker
    • HistoryTracker.plot()
    • HistoryTracker.register_debug_output_names()
    • HistoryTracker.update_debug_outputs()
    • HistoryTracker.update_history()
    • HistoryTracker.update_regression()
    • HistoryTracker.update_timeline()
  • Timer
    • Timer.t_passed
  • create_preview_batch_from_knossos()
  • pretty_string_time()
• elektronn3.training.trainer module
  • Backup
    • Backup.archive_backup()
  • NaNException
  • Trainer
    • Trainer.epoch
    • Trainer.exp_name
    • Trainer.out_channels
    • Trainer.run()
    • Trainer.save_path
    • Trainer.step
    • Trainer.tb
    • Trainer.terminate
    • Trainer.train_loader
    • Trainer.valid_loader
  • findcudatensors()
• elektronn3.training.trainer_gnn module
• elektronn3.training.trainer_gnn_batch module
• elektronn3.training.trainer_gnn_minibatch module
• elektronn3.training.triplettrainer module
  • TripletTrainer
    • TripletTrainer.epoch
    • TripletTrainer.exp_name
    • TripletTrainer.out_channels
    • TripletTrainer.save_path
    • TripletTrainer.step
    • TripletTrainer.tb
    • TripletTrainer.terminate
    • TripletTrainer.train_loader
    • TripletTrainer.valid_loader