Skip to content

feat: add Dropout to Flax UNet#3894

Merged
pcuenca merged 12 commits into
huggingface:mainfrom
SauravMaheshkar:saurav/add-dropout-flax-unet
Jul 7, 2023
Merged

feat: add Dropout to Flax UNet#3894
pcuenca merged 12 commits into
huggingface:mainfrom
SauravMaheshkar:saurav/add-dropout-flax-unet

fix: drop nn.compact

364432b
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs