Skip to content

[0.3.5] TensorNeko Release

Compare
Choose a tag to compare
@github-actions github-actions released this 04 Nov 12:37
· 102 commits to master since this 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, 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