Documentation

• Train Network
  • MainModule
  • Net
  • get_parser()
• Training sequential Networks
  • SequentialTrain
• Sequential Inference
  • SequentialPredictions
• Networks Architectures Package
  • Attention-UNet Architecture Module
  • Attention-UNet utils module
  • Layers module
  • MedFormer Architecture Module
  • MedFormer utils module
  • MedFormer utils-2 module
  • nnFormer Architecture Module
  • nnFormer utils module
  • SegFormer Architecture Module
  • SwinUNETR Architecture Module
  • trans-layers module
  • UNet Architecture Module
  • UNet utils module
  • UNet++ Architecture Module
  • UNETR Architecture Module
  • utils module
  • VNet Architecture Module
  • VTUNet Architecture Module
  • VTUNet utils module
  • Architecture selection module
• Profiler module
  • calculate_flops()
  • calculate_params()
  • estimate_memory_inference()
  • get_parser()
  • human_readable()
  • main()
  • main_profile()
  • memory_profile()
  • parse_memory_size()
  • separate_number_and_unit()
  • time_profile()
  • write_data_json()
• Charts module
• Sequential Measures to Networks
  • SequentialMetrics
• training package
  • Submodules
  • training.utils module
  • Module contents
• utils module
  • check_gpu_memory()
  • configure_logger()
  • get_latest_run_version_ckpt_epoch_no()
  • is_master()
  • sample_stack()
  • save_configure()
  • save_nifti()