Comments (2)
Hi @matthias-wright, sorry for the late response. The notebook was running fine on CPU runtime, but gave the above error on the GPU runtime (repeatedly). On checking again now, it seems to work well (reason unknown).
Thanks for building this library, helps a lot.
Closing the issue
from flaxmodels.
Hi @DevPranjal, I cannot reproduce this error. The notebook is working fine for me on Colab. What is the part of the code that caused this error for you?
Can you execute this command in the Colab and post the result here:
!ls /tmp/flaxmodels
from flaxmodels.
Related Issues (20)
- StyleGAN 2 doesn't work with Colab TPU despite successful TPU connection/init? HOT 9
- Using fixed versions in requirements.txt to prevent dependency resolver issues HOT 1
- resnet_demo.ipynb - JaxLib Version needs to be updated + Error HOT 1
- Question about batch norm HOT 2
- How did you convert pytorch model weight to Flax? HOT 2
- How to do transfer learning with flaxmodels? HOT 1
- unexpected keyword argument 'rng' in Dropout with new flax HOT 1
- Installation - Numpy version HOT 2
- PyPI distribution HOT 2
- How to train styleGAN on Colab TPU for custom dataset? HOT 2
- More info about the weight preprocessing step HOT 1
- StyleGAN2 `--fmap_base` only affects generator feature maps HOT 1
- Weight Initialization PRNGKey HOT 2
- Possible bug in StyleGAN training code HOT 1
- Use `safetensors` to store tensors instead of `pickle` HOT 3
- Remove head option in Resnet
- Hosting models on a more reliable service HOT 5
- Input preprocessing for VGG HOT 4
- `data_pipeline.py` needs more changes than suggested in README to support `ImageFolder` datasets HOT 5
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from flaxmodels.