Skip to content

Releases: ViCCo-Group/thingsvision

Bug fixes with TF

04 Dec 11:40
Compare
Choose a tag to compare
  • fixed TF preprocessing issues
  • now, Harmonized models work as expected

Bug fixes

11 Sep 15:08
Compare
Choose a tag to compare
  • Bug fixes regarding the SSL models and torch hub

New models: SAM and ALIGN

25 Jul 13:11
ad903f3
Compare
Choose a tag to compare

New models: SAM and ALIGN

27 May 08:38
ab01641
Compare
Choose a tag to compare
  • Enabled feature extraction for new models: SAM and Kakaobrain's ALIGN implementation
  • Fixed small bugs in rbf kernel of CKA class

PyTorch 2.0 support and CKA speedups + improvements

22 Apr 09:43
29c3370
Compare
Choose a tag to compare
  • added support for torch>=2.0.0 and torchvision=0.15.* (fixes issue #165)
  • implemented CKA in both NumPy and PyTorch (PyTorch runs 2x faster than NumPy)
  • added CUDA support for CKA in PyTorch (100x faster than CKA in NumPy)
  • added new function called get_cka where the user can set the backend (torch or numpy); function returns the CKA object
  • added an unbiased CKA variant (from here) in addition to vanilla CKA

Efficient batch-wise feature extraction + MAEs

08 Apr 11:22
f606f6a
Compare
Choose a tag to compare
  • improved the efficiency of batch-wise feature extraction (with custom data pipeline)
  • added pretrained MAEs to the thingsvision model collection

Efficient batch-wise feature extraction

05 Apr 13:16
9e6344b
Compare
Choose a tag to compare
  • efficient batch-wise feature extraction (using a with statement for PyTorch models) for custom data pipeline

Batch-wise feature extraction and human alignment

04 Apr 09:46
8223b61
Compare
Choose a tag to compare
  • added a public batch-wise extraction method to the feature extractor (for custom dataset and data loader classes)
  • updated README and docs with explanations about how and when to use the mini-batch extraction method
  • small bug fixes and refactorings

v2.5.0

28 Mar 18:05
Compare
Choose a tag to compare
  • added method to extractor class for aligning representations using gLocal
  • small bug fixes and refactoring

v2.4.2

19 Mar 10:02
Compare
Choose a tag to compare
  • new DreamSim models
  • support for CLS token extraction for supervised ViTs