PyPI page
Home page
Author:
None
License:
The extensive modifications made to the original code are released/relicensed under: GNU General Public License v3 Copyright (c) 2023, Andi Powers-Holmes This program is free software: you can redi...
Summary:
A neural network trainer (for weebs)
Latest version:
0.1.3
Required dependencies:
accelerate
|
chardet
|
colorama
|
colorcet
|
crc32c
|
diffusers
|
einops
|
fsspec
|
jsonargparse
|
kornia
|
lightning
|
natsort
|
ninja
|
numpy
|
open-clip-torch
|
opencv-contrib-python-headless
|
pandas
|
pillow
|
pudb
|
pydantic
|
pydantic-settings
|
pytorch-optimizer
|
rich
|
safetensors
|
scipy
|
sentencepiece
|
shellingham
|
tokenizers
|
torch
|
torchmetrics
|
torchvision
|
tqdm
|
transformers
|
typer
|
wandb
Optional dependencies:
adlfs
|
bitsandbytes
|
datasets
|
hf-transfer
|
neurosis
|
nvidia-ml-py
|
pre-commit
|
pyarrow
|
pymongoarrow
|
ruff
|
s3fs
|
setuptools-scm
|
tabulate
|
torchelastic
|
zstandard
Downloads last day:
0
Downloads last week:
6
Downloads last month:
17