PyPI - tensorneko - Versions diffs - 0.3.18__tar.gz → 0.3.20__tar.gz - Mend

tensorneko 0.3.18tar.gz → 0.3.20tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (100) hide show

{tensorneko-0.3.18 → tensorneko-0.3.20}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tensorneko
-Version: 0.3.18
+Version: 0.3.20
 Summary: Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lightning.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
@@ -22,7 +22,7 @@ Description: <h1 style="text-align: center">TensorNeko</h1>
         <div align="center">
             <a href="https://www.python.org/"><img src="https://img.shields.io/pypi/pyversions/tensorneko?style=flat-square"></a>
             <a href="https://pytorch.org/"><img src="https://img.shields.io/badge/PyTorch-%3E%3D1.9.0-EE4C2C?style=flat-square&logo=pytorch"></a>
-            <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.0.*%20|%202.1.*%20|%202.2.*-792EE5?style=flat-square&logo=lightning"></a>
+            <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.*-792EE5?style=flat-square&logo=lightning"></a>
         </div>
         <div align="center">

{tensorneko-0.3.18 → tensorneko-0.3.20}/README.md RENAMED Viewed

@@ -12,7 +12,7 @@
 <div align="center">
     <a href="https://www.python.org/"><img src="https://img.shields.io/pypi/pyversions/tensorneko?style=flat-square"></a>
     <a href="https://pytorch.org/"><img src="https://img.shields.io/badge/PyTorch-%3E%3D1.9.0-EE4C2C?style=flat-square&logo=pytorch"></a>
-    <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.0.*%20|%202.1.*%20|%202.2.*-792EE5?style=flat-square&logo=lightning"></a>
+    <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.*-792EE5?style=flat-square&logo=lightning"></a>
 </div>
 <div align="center">

{tensorneko-0.3.18 → tensorneko-0.3.20}/src/tensorneko/neko_model.py RENAMED Viewed

@@ -214,12 +214,12 @@ class NekoModel(LightningModule, NekoModule):
         for key in outputs[0].keys():
             getter = summarize_dict_by(key, torch.mean)
             value = getter(outputs)
-            self.history[-1]["val_" + key] = value
+            self.history[-1]["val_" + key.split("/")[-1]] = value
             if self.logger is not None:
                 self.logger.log_metrics({key: value}, step=self.trainer.global_step)
             self.log(key, value, on_epoch=True, on_step=False, logger=False, sync_dist=self.distributed)
-            self.log(f"val_{key}", value, on_epoch=True, on_step=False, logger=False, prog_bar=True,
-                sync_dist=self.distributed)
+            self.log(f"val_{key.split('/')[-1]}", value, on_epoch=True, on_step=False, logger=False,
+                prog_bar=True, sync_dist=self.distributed)
     def log_on_training_step_end(self, output: STEP_OUTPUT) -> None:
         """Log the training step outputs"""

{tensorneko-0.3.18 → tensorneko-0.3.20}/src/tensorneko/util/__init__.py RENAMED Viewed

@@ -4,7 +4,8 @@ from tensorneko_util.util import dispatch, AverageMeter, tensorneko_util_path
 from tensorneko_util.util.fp import Seq, AbstractSeq, curry, F, Stream, return_option, Option, Monad, Eval, _, __
 from tensorneko_util.util import ref, Timer, Singleton
 from tensorneko_util.util.eventbus import Event, EventBus, EventHandler, subscribe
-from tensorneko_util.util import download_file, download_file_thread, download_files_thread, WindowMerger, Registry
+from tensorneko_util.util import download_file, download_file_thread, download_files_thread, WindowMerger, Registry, \
+    MultiLayerIndexer
 from . import type
 from .configuration import Configuration
 from .misc import reduce_dict_by, summarize_dict_by, with_printed_shape, is_bad_num, count_parameters, compose, \
@@ -78,5 +79,6 @@ __all__ = [
     "download_files_thread",
     "WindowMerger",
     "Registry",
-    "run_gc"
+    "run_gc",
+    "MultiLayerIndexer"
 ]

tensorneko-0.3.20/src/tensorneko/version.txt ADDED Viewed

	@@ -0,0 +1 @@
1	+ 0.3.20

{tensorneko-0.3.18 → tensorneko-0.3.20}/src/tensorneko.egg-info/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tensorneko
-Version: 0.3.18
+Version: 0.3.20
 Summary: Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lightning.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
@@ -22,7 +22,7 @@ Description: <h1 style="text-align: center">TensorNeko</h1>
         <div align="center">
             <a href="https://www.python.org/"><img src="https://img.shields.io/pypi/pyversions/tensorneko?style=flat-square"></a>
             <a href="https://pytorch.org/"><img src="https://img.shields.io/badge/PyTorch-%3E%3D1.9.0-EE4C2C?style=flat-square&logo=pytorch"></a>
-            <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.0.*%20|%202.1.*%20|%202.2.*-792EE5?style=flat-square&logo=lightning"></a>
+            <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.*-792EE5?style=flat-square&logo=lightning"></a>
         </div>
         <div align="center">

{tensorneko-0.3.18 → tensorneko-0.3.20}/src/tensorneko.egg-info/requires.txt RENAMED Viewed

@@ -2,7 +2,7 @@ av>=8.0.3
 einops>=0.3.0
 numpy>=1.20.1
 pillow>=8.1
-tensorneko_util==0.3.18
+tensorneko_util==0.3.20
 torch>=1.9.0
 torchaudio>=0.9.0
 torchmetrics>=0.7.3
@@ -12,5 +12,5 @@ torchvision>=0.10.0
 pysoundfile>=0.9.0
 [lightning]
-lightning<2.2,>=2.0
+lightning<2.5,>=2.0
 tensorboard>=2.0.0