feat: add `Dropout` to Flax UNet #3894
pcuenca
merged
12 commits into
huggingface:main
from
saurav/add-dropout-flax-unet
feat: add Dropout to Flax UNet
b33274e3
Merge branch 'main' into saurav/add-dropout-flax-unet
2b18ad49
feat: add @compact decorator
d4da0991
Merge branch 'main' into saurav/add-dropout-flax-unet
d7e2c0cc
Merge branch 'main' into saurav/add-dropout-flax-unet
35e392b8
Merge branch 'main' into saurav/add-dropout-flax-unet
a259d56c
Merge branch 'main' into saurav/add-dropout-flax-unet
36f6181a
Merge branch 'main' into saurav/add-dropout-flax-unet
725e3c56
Merge branch 'main' into saurav/add-dropout-flax-unet
b4564efa
Merge branch 'main' into saurav/add-dropout-flax-unet
4dd27377
Merge branch 'main' into saurav/add-dropout-flax-unet
63ecf70e
fix: drop nn.compact
364432b5
pcuenca
approved these changes
on 2023-07-07
Login to write a write a comment.
Login via GitHub