tensorneko 0.3.16__tar.gz → 0.3.17__tar.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
  1. {tensorneko-0.3.16 → tensorneko-0.3.17}/PKG-INFO +1 -1
  2. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/__init__.py +2 -0
  3. tensorneko-0.3.17/src/tensorneko/util/gc.py +11 -0
  4. tensorneko-0.3.17/src/tensorneko/version.txt +1 -0
  5. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/PKG-INFO +1 -1
  6. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/SOURCES.txt +1 -0
  7. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/requires.txt +1 -1
  8. tensorneko-0.3.16/src/tensorneko/version.txt +0 -1
  9. {tensorneko-0.3.16 → tensorneko-0.3.17}/README.md +0 -0
  10. {tensorneko-0.3.16 → tensorneko-0.3.17}/setup.cfg +0 -0
  11. {tensorneko-0.3.16 → tensorneko-0.3.17}/setup.py +0 -0
  12. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/__init__.py +0 -0
  13. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/arch/__init__.py +0 -0
  14. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/arch/auto_encoder.py +0 -0
  15. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/arch/binary_classifier.py +0 -0
  16. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/arch/gan.py +0 -0
  17. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/arch/vqvae.py +0 -0
  18. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/arch/wgan.py +0 -0
  19. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/backend/__init__.py +0 -0
  20. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/__init__.py +0 -0
  21. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/display_metrics_callback.py +0 -0
  22. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/earlystop_lr.py +0 -0
  23. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/epoch_num_logger.py +0 -0
  24. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/epoch_time_logger.py +0 -0
  25. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/gpu_stats_logger.py +0 -0
  26. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/lr_logger.py +0 -0
  27. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/nil_callback.py +0 -0
  28. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/callback/system_stats_logger.py +0 -0
  29. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/dataset/__init__.py +0 -0
  30. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/dataset/list_dataset.py +0 -0
  31. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/dataset/nested_dataset.py +0 -0
  32. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/dataset/round_robin_dataset.py +0 -0
  33. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/dataset/sampler/__init__.py +0 -0
  34. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/dataset/sampler/sequential_iter_sampler.py +0 -0
  35. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/debug/__init__.py +0 -0
  36. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/__init__.py +0 -0
  37. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/enum.py +0 -0
  38. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/fid.py +0 -0
  39. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/iou.py +0 -0
  40. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/psnr.py +0 -0
  41. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/secs.py +0 -0
  42. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/evaluation/ssim.py +0 -0
  43. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/__init__.py +0 -0
  44. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/mesh/__init__.py +0 -0
  45. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/mesh/mesh_reader.py +0 -0
  46. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/mesh/mesh_writer.py +0 -0
  47. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/reader.py +0 -0
  48. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/weight/__init__.py +0 -0
  49. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/weight/weight_reader.py +0 -0
  50. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/weight/weight_writer.py +0 -0
  51. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/io/writer.py +0 -0
  52. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/__init__.py +0 -0
  53. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/aggregation.py +0 -0
  54. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/attention.py +0 -0
  55. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/concatenate.py +0 -0
  56. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/conv.py +0 -0
  57. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/linear.py +0 -0
  58. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/log.py +0 -0
  59. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/masked_conv2d.py +0 -0
  60. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/noise.py +0 -0
  61. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/patching.py +0 -0
  62. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/positional_embedding.py +0 -0
  63. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/reshape.py +0 -0
  64. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/stack.py +0 -0
  65. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/layer/vector_quantizer.py +0 -0
  66. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/__init__.py +0 -0
  67. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/dense.py +0 -0
  68. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/gated_conv.py +0 -0
  69. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/inception.py +0 -0
  70. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/mlp.py +0 -0
  71. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/residual.py +0 -0
  72. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/module/transformer.py +0 -0
  73. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/msg/__init__.py +0 -0
  74. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/neko_model.py +0 -0
  75. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/neko_module.py +0 -0
  76. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/neko_trainer.py +0 -0
  77. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/notebook/__init__.py +0 -0
  78. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/optim/__init__.py +0 -0
  79. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/optim/lr_scheduler/__init__.py +0 -0
  80. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/preprocess/__init__.py +0 -0
  81. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/preprocess/crop.py +0 -0
  82. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/preprocess/enum.py +0 -0
  83. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/preprocess/face_detector/__init__.py +0 -0
  84. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/preprocess/pad.py +0 -0
  85. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/preprocess/resize.py +0 -0
  86. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/configuration.py +0 -0
  87. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/dispatched_misc.py +0 -0
  88. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/misc.py +0 -0
  89. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/reproducibility.py +0 -0
  90. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/string_getter.py +0 -0
  91. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/type.py +0 -0
  92. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/visualization/__init__.py +0 -0
  93. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/visualization/image_browser/__init__.py +0 -0
  94. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/visualization/log_graph.py +0 -0
  95. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/visualization/matplotlib.py +0 -0
  96. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/visualization/watcher/__init__.py +0 -0
  97. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/dependency_links.txt +0 -0
  98. {tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/top_level.txt +0 -0
  99. {tensorneko-0.3.16 → tensorneko-0.3.17}/test/test_library_info.py +0 -0
  100. {tensorneko-0.3.16 → tensorneko-0.3.17}/test/test_version.py +0 -0
{tensorneko-0.3.16 → tensorneko-0.3.17}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tensorneko
3
- Version: 0.3.16
3
+ Version: 0.3.17
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
{tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko/util/__init__.py RENAMED
@@ -15,6 +15,7 @@ from .dispatched_misc import sparse2binary, binary2sparse
15
15
  from .reproducibility import Seed
16
16
  from .string_getter import get_activation, get_loss
17
17
  from .type import ModuleFactory, Shape, Device
18
+ from .gc import run_gc
18
19
 
19
20
  Rearrange = _Rearrange
20
21
 
@@ -77,4 +78,5 @@ __all__ = [
77
78
  "download_files_thread",
78
79
  "WindowMerger",
79
80
  "Registry",
81
+ "run_gc"
80
82
  ]
tensorneko-0.3.17/src/tensorneko/util/gc.py ADDED
@@ -0,0 +1,11 @@
1
+ # clear gc for python, pytorch, etc
2
+ import gc
3
+ import torch.cuda
4
+
5
+
6
+ def run_gc():
7
+ """
8
+ Clear memory for Python, PyTorch, etc.
9
+ """
10
+ gc.collect()
11
+ torch.cuda.empty_cache()
tensorneko-0.3.17/src/tensorneko/version.txt ADDED
@@ -0,0 +1 @@
1
+ 0.3.17
{tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tensorneko
3
- Version: 0.3.16
3
+ Version: 0.3.17
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
{tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/SOURCES.txt RENAMED
@@ -83,6 +83,7 @@ src/tensorneko/preprocess/face_detector/__init__.py
83
83
  src/tensorneko/util/__init__.py
84
84
  src/tensorneko/util/configuration.py
85
85
  src/tensorneko/util/dispatched_misc.py
86
+ src/tensorneko/util/gc.py
86
87
  src/tensorneko/util/misc.py
87
88
  src/tensorneko/util/reproducibility.py
88
89
  src/tensorneko/util/string_getter.py
{tensorneko-0.3.16 → tensorneko-0.3.17}/src/tensorneko.egg-info/requires.txt RENAMED
@@ -2,7 +2,7 @@ av>=8.0.3
2
2
  einops>=0.3.0
3
3
  numpy>=1.20.1
4
4
  pillow>=8.1
5
- tensorneko_util==0.3.16
5
+ tensorneko_util==0.3.17
6
6
  torch>=1.9.0
7
7
  torchaudio>=0.9.0
8
8
  torchmetrics>=0.7.3
tensorneko-0.3.16/src/tensorneko/version.txt DELETED
@@ -1 +0,0 @@
1
- 0.3.16
{tensorneko-0.3.16 → tensorneko-0.3.17}/README.md RENAMED
File without changes
{tensorneko-0.3.16 → tensorneko-0.3.17}/setup.cfg RENAMED
File without changes
{tensorneko-0.3.16 → tensorneko-0.3.17}/setup.py RENAMED
File without changes