Fix buggy loss - Githubissues

IgorSusmelj / barlowtwins

Implementation of Barlow Twins paper

MIT License

99 stars 12 forks source link

Fix buggy loss #1

Closed philippmwirth closed 3 years ago

philippmwirth commented 3 years ago

The off-diagonal elements were not really multiplied by the lambda_param. I fixed it and additionally upgraded the lightly requirement to 1.0.8. With this setup I'm at 56% accuracy after 48 epochs on CIFAR-10.