[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