Hi
I tried to rerun your code for cifar100 multi-class with the following command:
CUDA_VISIBLE_DEVICES=4,5,6,7 python -m torch.distributed.launch --nproc_per_node=4 train.py --dataset 'cifar100' --model 'resnet18' --mode sup_simclr_CSI --shift_trans_type rotation --batch_size 32 --epoch 700
But I got very poor results. Can you possibly provide me with a checkpoint for the same config if you have already trained any?
thanks