SupCon-Framework

SupCon-Framework copied to clipboard

Would you consider [adding a license](https://choosealicense.com/) to this repo? Otherwise it is [not clear](https://choosealicense.com/no-permission/) what other people are allowed to do with your code. Thanks!

dpaleka

ValueError in Stage2

3

Hi, I'm trying to run your code on CIFAR-10. The training and SWA in stage1 were fine, but I got the following error when training stage2: ``` shell root@864d7f9c24b4:/SupCon-Framework# python...

JiarunLiu

RT

scriptboy1990

Hi, your project SupCon-Framework requires "albumentations==0.5.2" in its dependency. After analyzing the source code, we found that the following versions of albumentations can also be suitable without affecting your project,...

Agnes-U

Dear Ivan, than you very much for your work! The output of the model in utils.py in line 250: output = model(data) is a tensor of [20,2] (the batch size...

karantai