pytorch-summary
livelossplot
Our great sponsors
pytorch-summary | livelossplot | |
---|---|---|
2 | 1 | |
3,957 | 1,282 | |
- | - | |
0.0 | 0.0 | |
about 2 months ago | almost 2 years ago | |
Python | Python | |
MIT License | MIT License |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
pytorch-summary
-
Conv3D
could try this package, pytorch-summary package if you havent
-
Why is CUDA running out of memory?
This could be because of session thats is running in background which you need to terminate and refresh.Or it might be due to large nn model so maybe try changing the number of layers and neurons in the model.This summary library might help you check the size of your nn model
livelossplot
What are some alternatives?
horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
best-of-ml-python - 🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
wandb - 🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
tf-keras-vis - Neural network visualization toolkit for tf.keras
netron - Visualizer for neural network, deep learning and machine learning models
einops - Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
pytorch-model-details - Package to visualize all the layer parameters of a deep learning model.
horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet. [Moved to: https://github.com/horovod/horovod]
d2l-en - Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.