vicregaddon 0.0.4

A lightweight and modular parallel PyTorch implementation of VICReg (intended for audio, but will try to be general)

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)

Installation:

pip install vicregaddon

or, for the most recent version

pip install git+https://github.com/drscotthawley/vicregaddon.git

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.