PyPI page
Home page
Author:
None
License:
MIT
Summary:
A Python package for estimating GPU memory requirements and the number of GPUs needed for training machine learning models
Latest version:
0.2.0
Required dependencies:
click
|
huggingface_hub
|
numpy
|
torch
|
transformers
Optional dependencies:
black
|
flake8
|
pytest
|
pytest-cov
Downloads last day:
0
Downloads last week:
8
Downloads last month:
13