A lightweight and modular parallel PyTorch implementation of VICReg (intended for audio, but will try to be general)
Project description
vicregaddon
"vicreg add-on". As in "Yeah, but are you getting it? vicregaddon it"
A lightweight and modular parallel PyTorch implementation of VICReg.
(I intend this for audio, but could be used for other things)
Usage:
from vicregaddon import *
Dependencies:
- torch
- einops
In the library itself, parallelism is handled directly via torch.distributed and thus is "agnostic" to whether PyTorch Lightning or Accelerate (or some other thing) is used.
The examples/ however, have several more dependencies. You will need to install Lightning and/or Accelerate and/or other things to run them, depending on each example.
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file vicregaddon-0.0.2.tar.gz.
File metadata
- Download URL: vicregaddon-0.0.2.tar.gz
- Upload date:
- Size: 6.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.11.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 56270e64b0f28667712d52ae66df3e30fcec3cacff467b6f8b56ef1c6ed1ee8c |
|
| MD5 | d642755bb47b07a1cc1906d4e0df938a |
|
| BLAKE2b-256 | 9bf77d7765a23bd47f80352f2c341d6149ad406ded8ebf0f8378113420a44fd0 |
