livelossplot
pytorch-summary
livelossplot | pytorch-summary | |
---|---|---|
1 | 2 | |
1,282 | 3,957 | |
- | - | |
0.0 | 0.0 | |
almost 2 years ago | 2 months 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.
livelossplot
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
What are some alternatives?
best-of-ml-python - 🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
tf-keras-vis - Neural network visualization toolkit for tf.keras
wandb - 🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
einops - Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
netron - Visualizer for neural network, deep learning and machine learning models
horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet. [Moved to: https://github.com/horovod/horovod]
pytorch-model-details - Package to visualize all the layer parameters of a deep learning model.
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.