VICReg icon indicating copy to clipboard operation
VICReg copied to clipboard

Published 20 hours ago • verified publisher icon AnnaManasyan
→

Metadata

An easy-to-use implementation of VICReg in Pytorch.

Results 1 VICReg issues
Sort by recently updated
recently updated
newest added

Does this code need large GPU memory？

1 comment

Hello! During the training of cifar10 dataset, do you encounter that when the batchsize is set to 2048, you can't run on the dual card nvidia3090? Display memory overflow. So...

bqdeng avatar bqdeng

← Metadata

19
Stars
7
Forks
Watchers

Owner

verified publisher icon AnnaManasyan

Metadata

An easy-to-use implementation of VICReg in Pytorch.

Back