pytinytensor 0.1.3__tar.gz → 0.1.4__tar.gz

{pytinytensor-0.1.3/pytinytensor.egg-info → pytinytensor-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytinytensor
-Version: 0.1.3
+Version: 0.1.4
 Summary: мини ИИ фреймворк от IbrokimN ( github/IbrokhimN )
 Home-page: https://github.com/IbrokhimN/tinytensor
 Author: IbrokhimN
@@ -33,6 +33,7 @@ Dynamic: summary
 ## Содержание
 
 - [Установка](#установка)
+- [Документация](#документация)
 - [Быстрый старт](#быстрый-старт)
 - [Как работает autograd](#как-работает-autograd)
 - [API](#api)
@@ -59,6 +60,20 @@ pip install -r requirements.txt
 
 Зависимость одна - numpy.
 
+## Документация
+
+Все подробности разложены по [`docs/`](docs/), README тут больше как входная дверь:
+
+- [docs/getting_started.md](docs/getting_started.md) - установка, quickstart
+- [docs/tensor_and_autograd.md](docs/tensor_and_autograd.md) - как устроен Tensor и backward
+- [docs/nn.md](docs/nn.md) - Module, Linear, активации, Dropout, MSELoss
+- [docs/optim.md](docs/optim.md) - SGD, AdamW
+- [docs/data.md](docs/data.md) - Dataset, DataLoader
+- [docs/utils.md](docs/utils.md) - progress_bar, EarlyStopping
+- [docs/model_saving.md](docs/model_saving.md) - save/load модели
+- [docs/cuda.md](docs/cuda.md) - опциональный cuda-бэкенд
+- [docs/faq.md](docs/faq.md) - грабли, на которые уже наступили
+
 ## Быстрый старт
 
 ```python

{pytinytensor-0.1.3 → pytinytensor-0.1.4}/README.md

@@ -10,6 +10,7 @@
 ## Содержание
 
 - [Установка](#установка)
+- [Документация](#документация)
 - [Быстрый старт](#быстрый-старт)
 - [Как работает autograd](#как-работает-autograd)
 - [API](#api)
@@ -36,6 +37,20 @@ pip install -r requirements.txt
 
 Зависимость одна - numpy.
 
+## Документация
+
+Все подробности разложены по [`docs/`](docs/), README тут больше как входная дверь:
+
+- [docs/getting_started.md](docs/getting_started.md) - установка, quickstart
+- [docs/tensor_and_autograd.md](docs/tensor_and_autograd.md) - как устроен Tensor и backward
+- [docs/nn.md](docs/nn.md) - Module, Linear, активации, Dropout, MSELoss
+- [docs/optim.md](docs/optim.md) - SGD, AdamW
+- [docs/data.md](docs/data.md) - Dataset, DataLoader
+- [docs/utils.md](docs/utils.md) - progress_bar, EarlyStopping
+- [docs/model_saving.md](docs/model_saving.md) - save/load модели
+- [docs/cuda.md](docs/cuda.md) - опциональный cuda-бэкенд
+- [docs/faq.md](docs/faq.md) - грабли, на которые уже наступили
+
 ## Быстрый старт
 
 ```python

{pytinytensor-0.1.3 → pytinytensor-0.1.4/pytinytensor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytinytensor
-Version: 0.1.3
+Version: 0.1.4
 Summary: мини ИИ фреймворк от IbrokimN ( github/IbrokhimN )
 Home-page: https://github.com/IbrokhimN/tinytensor
 Author: IbrokhimN
@@ -33,6 +33,7 @@ Dynamic: summary
 ## Содержание
 
 - [Установка](#установка)
+- [Документация](#документация)
 - [Быстрый старт](#быстрый-старт)
 - [Как работает autograd](#как-работает-autograd)
 - [API](#api)
@@ -59,6 +60,20 @@ pip install -r requirements.txt
 
 Зависимость одна - numpy.
 
+## Документация
+
+Все подробности разложены по [`docs/`](docs/), README тут больше как входная дверь:
+
+- [docs/getting_started.md](docs/getting_started.md) - установка, quickstart
+- [docs/tensor_and_autograd.md](docs/tensor_and_autograd.md) - как устроен Tensor и backward
+- [docs/nn.md](docs/nn.md) - Module, Linear, активации, Dropout, MSELoss
+- [docs/optim.md](docs/optim.md) - SGD, AdamW
+- [docs/data.md](docs/data.md) - Dataset, DataLoader
+- [docs/utils.md](docs/utils.md) - progress_bar, EarlyStopping
+- [docs/model_saving.md](docs/model_saving.md) - save/load модели
+- [docs/cuda.md](docs/cuda.md) - опциональный cuda-бэкенд
+- [docs/faq.md](docs/faq.md) - грабли, на которые уже наступили
+
 ## Быстрый старт
 
 ```python

{pytinytensor-0.1.3 → pytinytensor-0.1.4}/pytinytensor.egg-info/SOURCES.txt

@@ -37,4 +37,5 @@ tinytensor/optim/__init__.py
 tinytensor/optim/optimizer.py
 tinytensor/utils/__init__.py
 tinytensor/utils/bar.py
-tinytensor/utils/early_stopping.py
+tinytensor/utils/early_stopping.py
+tinytensor/utils/summary.py

{pytinytensor-0.1.3 → pytinytensor-0.1.4}/setup.py

@@ -113,7 +113,7 @@ else:
 
 setup(
     name="pytinytensor",
-    version="0.1.3",
+    version="0.1.4",
     description="мини ИИ фреймворк от IbrokimN ( github/IbrokhimN )",
     long_description=long_description,
     long_description_content_type="text/markdown",

{pytinytensor-0.1.3 → pytinytensor-0.1.4}/tinytensor/utils/__init__.py

@@ -1,4 +1,5 @@
 from tinytensor.utils.bar import progress_bar, train_bar
 from tinytensor.utils.early_stopping import EarlyStopping
-__all__ = ["progress_bar", "train_bar",
-           "EarlyStopping"]
+from tinytensor.utils.summary import summary
+
+__all__ = ["progress_bar", "train_bar", "EarlyStopping", "summary"]

pytinytensor-0.1.4/tinytensor/utils/summary.py

@@ -0,0 +1,49 @@
+import numpy as np
+from tinytensor.core.tensor import Tensor
+from tinytensor.nn.modules import Module 
+
+def summary(model: Module, input_shape: tuple):
+    # печает инфу о модели
+    print("\n" + "=" * 50)
+    print(f"{'layer (type)':<20} {'output shape':<18} {'param #':<10}")
+    print("-" * 50)
+
+    total_params = 0
+    trainable_params = 0
+
+    dummy_input = Tensor(np.zeros(input_shape), requires_grad=False)
+    x = dummy_input
+
+    layers = []
+    for attr in model.__dict__.values():
+        if isinstance(attr, Module):
+            layers.append(attr)
+        elif isinstance(attr, (list, tuple)):
+            for item in attr:
+                if isinstance(item, Module):
+                    layers.append(item)
+
+    for layer in layers:
+        layer_name = layer.__class__.__name__
+
+        layer_params = 0
+        for p in layer.parameters():
+            num_params = int(np.prod(p.data.shape))
+            layer_params += num_params
+            if p.requires_grad:
+                trainable_params += num_params
+            total_params += num_params
+
+        try:
+            x = layer(x)
+            out_shape = str(tuple(x.data.shape))
+        except Exception as e:
+            out_shape = "Ошибка"
+
+        print(f"{layer_name:<20} {out_shape:<18} {layer_params:<10,}")
+
+    print("-" * 50)
+    print(f"Total params: {total_params:,}")
+    print(f"Trainable params: {trainable_params:,}")
+    print(f"Non-trainable params: {total_params - trainable_params:,}")
+    print("=" * 50 + "\n")