Releases: ControlNet/tensorneko
Releases · ControlNet/tensorneko
[0.3.11] TensorNeko Release
Features:
Singleton
support with class with init args- More utils for download files
- Typing for
Registry
[0.3.10] TensorNeko Release
Features:
Registry
for easier decorator-style class/func registrationread
andwrite
are also in the top level oftensorneko_util
[0.3.9] TensorNeko Release
Features:
JsonReader
now can load jsonl filesWeightReader
andWeightWriter
that can read and write thept
,ckpt
,safetensors
filesread
andwrite
now is in the top level of thetensorneko
packageimport_tqdm
andimport_tqdm_auto
now are not private
[0.3.8] TensorNeko Release
Features:
- Group the metrics, losses, and system stats in logger
- Add
input_shape
arg forBinaryClassifier
Fix:
NekoModel
cannot train when the logger isNone
[0.3.7] TensorNeko Release
Fix:
- Training error when using
lightning
2.1 - Docs are not clear for
VideoReader
,RoundRobinDataset
,NestedDataset
,ListDataset
.
[0.3.6] TensorNeko Release
Features:
- Better type hint for eventbus and dispatch
- Dynamic import for
mita_tqdm
on_step
andon_epoch
arguments forGpuStatsLogger
andSystemStatsLogger
Timer
now doesn't require parentheses as decorator
Fix:
crop_with_padding
using numpy arraysample_indexes
not importedFID
not working withprogress_bar=True
- import error with pytorch lightning 2.1.3
[0.3.5] TensorNeko Release
Features:
- Add support for pytorch 2.1 and lightning 2.1
- Auto infer for
notebook.display
, now supports audio, video, image, and code. ListDataset
, likeTensorDataset
intorch
but directly uses listSequentialIterSampler
, 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 setnum_workers > 0
.iou_2d
calculation error
[0.3.4] TensorNeko Release
Features:
Timer
now can disable the auto print, for better manual control.Read
andWrite
auto infer formattry_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
Features:
- Update
eventbus
api for less import. load_py
function inutil.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
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