tensorneko 0.3.19__py3-none-any.whl → 0.3.20__py3-none-any.whl

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 (7) hide show
  1. tensorneko/neko_model.py +3 -3
  2. tensorneko/version.txt +1 -1
  3. {tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/METADATA +4 -4
  4. {tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/RECORD +7 -7
  5. {tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/LICENSE +0 -0
  6. {tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/WHEEL +0 -0
  7. {tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/top_level.txt +0 -0
tensorneko/neko_model.py CHANGED
@@ -214,12 +214,12 @@ class NekoModel(LightningModule, NekoModule):
214
214
  for key in outputs[0].keys():
215
215
  getter = summarize_dict_by(key, torch.mean)
216
216
  value = getter(outputs)
217
- self.history[-1]["val_" + key] = value
217
+ self.history[-1]["val_" + key.split("/")[-1]] = value
218
218
  if self.logger is not None:
219
219
  self.logger.log_metrics({key: value}, step=self.trainer.global_step)
220
220
  self.log(key, value, on_epoch=True, on_step=False, logger=False, sync_dist=self.distributed)
221
- self.log(f"val_{key}", value, on_epoch=True, on_step=False, logger=False, prog_bar=True,
222
- sync_dist=self.distributed)
221
+ self.log(f"val_{key.split('/')[-1]}", value, on_epoch=True, on_step=False, logger=False,
222
+ prog_bar=True, sync_dist=self.distributed)
223
223
 
224
224
  def log_on_training_step_end(self, output: STEP_OUTPUT) -> None:
225
225
  """Log the training step outputs"""
tensorneko/version.txt CHANGED
@@ -1 +1 @@
1
- 0.3.19
1
+ 0.3.20
{tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tensorneko
3
- Version: 0.3.19
3
+ Version: 0.3.20
4
4
  Summary: Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lightning.
5
5
  Home-page: https://github.com/ControlNet/tensorneko
6
6
  Author: ControlNet
@@ -27,14 +27,14 @@ Requires-Dist: av (>=8.0.3)
27
27
  Requires-Dist: einops (>=0.3.0)
28
28
  Requires-Dist: numpy (>=1.20.1)
29
29
  Requires-Dist: pillow (>=8.1)
30
- Requires-Dist: tensorneko-util (==0.3.19)
30
+ Requires-Dist: tensorneko-util (==0.3.20)
31
31
  Requires-Dist: torch (>=1.9.0)
32
32
  Requires-Dist: torchaudio (>=0.9.0)
33
33
  Requires-Dist: torchmetrics (>=0.7.3)
34
34
  Requires-Dist: torchvision (>=0.10.0)
35
35
  Requires-Dist: pysoundfile (>=0.9.0) ; platform_system == "Windows"
36
36
  Provides-Extra: lightning
37
- Requires-Dist: lightning (<2.2,>=2.0) ; extra == 'lightning'
37
+ Requires-Dist: lightning (<2.5,>=2.0) ; extra == 'lightning'
38
38
  Requires-Dist: tensorboard (>=2.0.0) ; extra == 'lightning'
39
39
 
40
40
  <h1 style="text-align: center">TensorNeko</h1>
@@ -51,7 +51,7 @@ Requires-Dist: tensorboard (>=2.0.0) ; extra == 'lightning'
51
51
  <div align="center">
52
52
  <a href="https://www.python.org/"><img src="https://img.shields.io/pypi/pyversions/tensorneko?style=flat-square"></a>
53
53
  <a href="https://pytorch.org/"><img src="https://img.shields.io/badge/PyTorch-%3E%3D1.9.0-EE4C2C?style=flat-square&logo=pytorch"></a>
54
- <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>
54
+ <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.*-792EE5?style=flat-square&logo=lightning"></a>
55
55
  </div>
56
56
 
57
57
  <div align="center">
{tensorneko-0.3.19.dist-info → tensorneko-0.3.20.dist-info}/RECORD RENAMED
@@ -1,8 +1,8 @@
1
1
  tensorneko/__init__.py,sha256=uh1HNn1sNpX1bbOqAE_kNJfrH4eMtEzus0hO-Fh9tEw,990
2
- tensorneko/neko_model.py,sha256=QTbdOAg9ki0ix6mDL_Qu8Wmd5WJOoUFF3M1SXEp3KGc,10551
2
+ tensorneko/neko_model.py,sha256=hUMi7puzxW_6FOpA1jiFN1__oO5DZPlhhp3WXqhJXgg,10581
3
3
  tensorneko/neko_module.py,sha256=qELXvguSjWo_NvcRQibiFl0Qauzd9JWLSnT4dbGNS3Y,1473
4
4
  tensorneko/neko_trainer.py,sha256=JC8qoKSZ5ngz3grf3S0SjvIFVktDIP_GExth5aFfbGA,10074
5
- tensorneko/version.txt,sha256=xV2o_U6zdvujab1PTmKLrZ6dQjt1fW9CyXhxWiQJBpo,6
5
+ tensorneko/version.txt,sha256=QQxQgaQNGgPrr3bVpz_0fCx9e0SwIa-EL3bqC9TFcHE,6
6
6
  tensorneko/arch/__init__.py,sha256=w4lTUeyBIZelrnSjlBFWUF0erzOmBFl9FqeWQuSOyKs,248
7
7
  tensorneko/arch/auto_encoder.py,sha256=j6PWWyaNYaYNtw_zZ9ikzhCASqe9viXR3JGBIXSK92Y,2137
8
8
  tensorneko/arch/binary_classifier.py,sha256=1MkEbReXKLdDksRG5Rsife40grJk08EVDcNKp54Xvb4,2316
@@ -86,8 +86,8 @@ tensorneko/visualization/log_graph.py,sha256=NvOwWVc_petXWYdgaHosPFLa43sHBeacbYc
86
86
  tensorneko/visualization/matplotlib.py,sha256=xs9Ssc44ojZX65QU8-fftA7Ug_pBuZ3TBtM8vETNq9w,1568
87
87
  tensorneko/visualization/image_browser/__init__.py,sha256=AtykhAE3bXQS6SOWbeYFeeUE9ts9XOFMvrL31z0LoMg,63
88
88
  tensorneko/visualization/watcher/__init__.py,sha256=Nq752qIYvfRUZ8VctKQRSqhxh5KmFbWcqPfZlijVx6s,379
89
- tensorneko-0.3.19.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
90
- tensorneko-0.3.19.dist-info/METADATA,sha256=9M81K0ENnhym0S-qyKQTlQzSADnpOZIszNWJ66q4IuM,19998
91
- tensorneko-0.3.19.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
92
- tensorneko-0.3.19.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
93
- tensorneko-0.3.19.dist-info/RECORD,,
89
+ tensorneko-0.3.20.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
90
+ tensorneko-0.3.20.dist-info/METADATA,sha256=8namXF_Slh-e8MZDm-v0t3YhURZZyXHng6VSQ4H1d8E,19972
91
+ tensorneko-0.3.20.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
92
+ tensorneko-0.3.20.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
93
+ tensorneko-0.3.20.dist-info/RECORD,,