Skip to content

Releases: ControlNet/tensorneko

[0.3.11] TensorNeko Release

17 Mar 12:43
Compare
Choose a tag to compare

Features:

  • Singleton support with class with init args
  • More utils for download files
  • Typing for Registry

[0.3.10] TensorNeko Release

10 Mar 11:28
Compare
Choose a tag to compare

Features:

  • Registry for easier decorator-style class/func registration
  • read and write are also in the top level of tensorneko_util

[0.3.9] TensorNeko Release

19 Feb 13:30
Compare
Choose a tag to compare

Features:

  • JsonReader now can load jsonl files
  • WeightReader and WeightWriter that can read and write the pt, ckpt, safetensors files
  • read and write now is in the top level of the tensorneko package
  • import_tqdm and import_tqdm_auto now are not private

[0.3.8] TensorNeko Release

15 Feb 05:49
Compare
Choose a tag to compare

Features:

  • Group the metrics, losses, and system stats in logger
  • Add input_shape arg for BinaryClassifier

Fix:

  • NekoModel cannot train when the logger is None

[0.3.7] TensorNeko Release

03 Feb 09:34
Compare
Choose a tag to compare

Fix:

  • Training error when using lightning 2.1
  • Docs are not clear for VideoReader, RoundRobinDataset, NestedDataset, ListDataset.

[0.3.6] TensorNeko Release

09 Jan 18:16
Compare
Choose a tag to compare

Features:

  • Better type hint for eventbus and dispatch
  • Dynamic import for mita_tqdm
  • on_step and on_epoch arguments for GpuStatsLogger and SystemStatsLogger
  • Timer now doesn't require parentheses as decorator

Fix:

  • crop_with_padding using numpy array
  • sample_indexes not imported
  • FID not working with progress_bar=True
  • import error with pytorch lightning 2.1.3

[0.3.5] TensorNeko Release

04 Nov 12:37
Compare
Choose a tag to compare

Features:

  • Add support for pytorch 2.1 and lightning 2.1
  • Auto infer for notebook.display, now supports audio, video, image, and code.
  • ListDataset, like TensorDataset in torch but directly uses list
  • SequentialIterSampler, can split dataset to multiple epochs. Normally can be used for training in large-scale dataset but want to utilize the learning rate scheduler based on epochs.
  • BinaryClassifier. A predefined model for binary classification.
  • sample_indexes. Sample a continious indexes giving a source sequence length.

Fix:

  • Import problem raise by RoundRobinDataset
  • FID calculation problem when set num_workers > 0.
  • iou_2d calculation error

[0.3.4] TensorNeko Release

14 Oct 18:19
Compare
Choose a tag to compare

Features:

  • Timer now can disable the auto print, for better manual control.
  • Read and Write auto infer format
  • try_until_success for trying a function with multiple times if it raises exceptions.
  • DummyLogger for disabling annoy logging information when debug

Fix:

  • The unusual argument order in MatWriter

[0.3.3] TensorNeko Release

24 Sep 05:56
Compare
Choose a tag to compare

Features:

  • Update eventbus api for less import.
  • load_py function in util.misc for easiler import a py file as a module.

Fix:

  • Train loss is not displayed in the progress bar
  • Default audio backend doesn't work for audio writer and reader
  • FFmpeg -ss and -i wrong order for clipping videos
  • Add warn for if the input of video writer is from 0~255 unit8

[0.3.2] TensorNeko Release

02 Aug 12:28
Compare
Choose a tag to compare

Features:

  • Add ffmpeg backend for audio reader

Fix:

  • Tensorboard graph logging not working
  • Training will fail in DDP due to a bug in gpu logging callback