matthias-wright/flaxmodels
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Python
Issues
- 0
Remove head option in Resnet
#39 opened by vyeevani - 3
- 1
Possible bug in StyleGAN training code
#31 opened by wilson1yan - 1
- 1
More info about the weight preprocessing step
#28 opened by icysapphire - 2
- 2
PyPI distribution
#26 opened by pcuenca - 2
Installation - Numpy version
#25 opened by phlippe - 1
- 9
StyleGAN 2 doesn't work with Colab TPU despite successful TPU connection/init?
#11 opened by josephrocca - 1
How to do transfer learning with flaxmodels?
#20 opened by riven314 - 2
- 2
Resnet18 demo gives `OSError: Unable to open file (file signature not found)`
#16 opened by DevPranjal - 2
Question about batch norm
#15 opened by crawles - 1
- 1
Using fixed versions in requirements.txt to prevent dependency resolver issues
#13 opened by josephrocca - 5
`data_pipeline.py` needs more changes than suggested in README to support `ImageFolder` datasets
#8 opened by josephrocca - 4
Input preprocessing for VGG
#5 opened by borisdayma - 5
Hosting models on a more reliable service
#4 opened by borisdayma - 2
Weight Initialization PRNGKey
#3 opened by moabarar - 1
- 1
Links to old Colabs on Drive
#1 opened by andsteing