Fábio S. Ferreira

icon company University College London icon location London, UK

Results 1 issues of Fábio S. Ferreira

feat: add activation checkpointing to unet

4 comment

### Description Introduces an optional `use_checkpointing` flag in the `UNet` implementation. When enabled, intermediate activations in the encoder–decoder blocks are recomputed during the backward pass instead of being stored in...