Repository navigation

feat: add activation checkpointing to unet - #8554

Merged

ericspod merged 25 commits into

Project-MONAI:devfrom

ferreirafabio80:feat/add_activation_checkpointing_to_unet

Nov 14, 2025

Merged

feat: add activation checkpointing to unet#8554
ericspod merged 25 commits into
Project-MONAI:devfrom
ferreirafabio80:feat/add_activation_checkpointing_to_unet

Commits on Sep 3, 2025
- feat: add optional gradient checkpointing to unet
  Fabio Ferreira
  committed
  de2b6bd
- fix: small ruff issue
  Fabio Ferreira
  committed
  66edcb5
Commits on Sep 4, 2025
- Update monai/networks/nets/unet.py
  
  ferreirafabio80
  and
  coderabbitai[bot]
  authored
  e66e357
- docs: update docstrings
  Fabio Ferreira
  committed
  feefcaa
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  e112457
- fix: avoid BatchNorm subblocks
  Fabio Ferreira
  committed
  f673ca1
- fix: revert batch norm changes
  Fabio Ferreira
  committed
  69540ff
Commits on Oct 1, 2025
- refactor: creates a subclass of UNet and overrides the get connection block method
  Fabio Ferreira
  committed
  42ec757
- chore: remove use checkpointing from doc string
  Fabio Ferreira
  committed
  a2e8474
Commits on Oct 2, 2025
- fix: linting issues
  Fabio Ferreira
  committed
  4c4782e
Commits on Oct 8, 2025
- feat: add activation checkpointing to down and up paths to be more efficient
  Fabio Ferreira
  committed
  515c659
Commits on Nov 4, 2025
- refactor: move activation checkpointing wrapper to blocks
  Fabio Ferreira
  committed
  da5a3a4
- chore: add docstrings to checkpointed unet
  Fabio Ferreira
  committed
  43dec88
Commits on Nov 7, 2025
- test: add checkpoint unet test
  Fabio Ferreira
  committed
  84c0f48
- fix: change test name
  Fabio Ferreira
  committed
  5805515
- fix: simplify test and make sure that checkpoint unet runs well in training
  Fabio Ferreira
  committed
  1aa8e3c
- fix: set seed
  Fabio Ferreira
  committed
  447d9f2
- fix: fix testing bugs
  Fabio Ferreira
  committed
  b20a19e
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  41f000f
Commits on Nov 10, 2025
- chore: add test docstrings
  Fabio Ferreira
  committed
  a068c0e
- DCO Remediation Commit for Fabio Ferreira <f.ferreira@qureight.com>
  Fabio Ferreira
  committed
  26668cd
Commits on Nov 13, 2025
- fix: remove test script save
  Fabio Ferreira
  committed
  814fa80
- fix: tighten tolerance for numerical equivalence
  Fabio Ferreira
  committed
  c45ee48
Commits on Nov 14, 2025
- chore: update doc strings
  Fabio Ferreira
  committed
  4349d3f
- Merge branch 'dev' into feat/add_activation_checkpointing_to_unet
  KumoLiu
  authored
  885993b

Commits

Commits on Sep 3, 2025

Commits on Sep 4, 2025

Commits on Oct 1, 2025

Commits on Oct 2, 2025

Commits on Oct 8, 2025

Commits on Nov 4, 2025

Commits on Nov 7, 2025

Commits on Nov 10, 2025

Commits on Nov 13, 2025

Commits on Nov 14, 2025