Releases: ViCCo-Group/thingsvision
Releases · ViCCo-Group/thingsvision
Bug fixes with TF
Bug fixes
New models: SAM and ALIGN
- bug fixes
New models: SAM and ALIGN
PyTorch 2.0 support and CKA speedups + improvements
- added support for
torch>=2.0.0
andtorchvision=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 thebackend
(torch
ornumpy
); function returns the CKA object - added an unbiased CKA variant (from here) in addition to vanilla CKA
Efficient batch-wise feature extraction + MAEs
- 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
- efficient batch-wise feature extraction (using a
with
statement for PyTorch models) for custom data pipeline
Batch-wise feature extraction and human alignment
- 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