CSI icon indicating copy to clipboard operation
CSI copied to clipboard

Reproducing results for Cifar100 ens multi-class

Open nimaous opened this issue 4 years ago • 0 comments

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

nimaous avatar Nov 16 '21 12:11 nimaous