Our great sponsors
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
Having 53760 neurons takes much memory. Try adding more Conv2D layers or play with stride. Also, try .detach() to data and labels after training. Lastly, I would suggest to take a look at https://github.com/rentruewang/koila. Have not tried yet but it should be helpful.
NOTE:
The number of mentions on this list indicates mentions on common posts plus user suggested alternatives.
Hence, a higher number means a more popular project.
Related posts
- [D] Would the 8gb VRAM of the 3060ti mean that some models in computer vision cannot be trained with it at all?
- [P] Dynamic batching for GPT-J API
- Koila: Prevent PyTorch's out of memory error with lazy evaluation
- Solve PyTorch's `CUDA error: out of memory` in 1 line of code
- Can I use machine learning to vectorize floor plans?