├── README.md ├── cifar ├── analysis │ └── fix_fe.ipynb ├── aug │ ├── autoaug.py │ ├── cuda.py │ ├── cutmix.py │ ├── cutout.py │ ├── others.py │ ├── randaug.py │ └── transforms.py ├── datasets │ └── cifar100.py ├── losses │ ├── bcl.py │ ├── bs.py │ ├── ce.py │ ├── ce_drw.py │ ├── ldam_drw.py │ ├── loss.py │ ├── ncl.py │ └── ride.py ├── main.py ├── models │ ├── net.py │ ├── resnet.py │ ├── resnet_bcl.py │ ├── resnet_ncl.py │ └── resnet_ride.py ├── train │ ├── train.py │ ├── train_fn │ │ ├── base.py │ │ ├── bcl.py │ │ ├── ncl.py │ │ └── ride.py │ └── validate.py └── utils │ ├── accuracy.py │ ├── common.py │ ├── config.py │ ├── logger.py │ └── plot.py ├── large_scale ├── aug │ ├── cuda.py │ ├── randaug.py │ └── transforms.py ├── config │ └── iNaturalist │ │ └── iNaturalist18_val.txt ├── imagenet_train.py ├── imbalance_data │ ├── cmo.py │ ├── cutout.py │ ├── cv_aug.py │ └── lt_data.py ├── inat18_train.py ├── losses.py ├── opts.py ├── test.py ├── train.py ├── training_functions.py └── util │ ├── autoaug.py │ ├── moco_loader.py │ ├── randaugment.py │ └── util.py └── requirements.txt /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/README.md -------------------------------------------------------------------------------- /cifar/analysis/fix_fe.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/analysis/fix_fe.ipynb -------------------------------------------------------------------------------- /cifar/aug/autoaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/autoaug.py -------------------------------------------------------------------------------- /cifar/aug/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/cuda.py -------------------------------------------------------------------------------- /cifar/aug/cutmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/cutmix.py -------------------------------------------------------------------------------- /cifar/aug/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/cutout.py -------------------------------------------------------------------------------- /cifar/aug/others.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/others.py -------------------------------------------------------------------------------- /cifar/aug/randaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/randaug.py -------------------------------------------------------------------------------- /cifar/aug/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/aug/transforms.py -------------------------------------------------------------------------------- /cifar/datasets/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/datasets/cifar100.py -------------------------------------------------------------------------------- /cifar/losses/bcl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/bcl.py -------------------------------------------------------------------------------- /cifar/losses/bs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/bs.py -------------------------------------------------------------------------------- /cifar/losses/ce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/ce.py -------------------------------------------------------------------------------- /cifar/losses/ce_drw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/ce_drw.py -------------------------------------------------------------------------------- /cifar/losses/ldam_drw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/ldam_drw.py -------------------------------------------------------------------------------- /cifar/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/loss.py -------------------------------------------------------------------------------- /cifar/losses/ncl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/ncl.py -------------------------------------------------------------------------------- /cifar/losses/ride.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/losses/ride.py -------------------------------------------------------------------------------- /cifar/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/main.py -------------------------------------------------------------------------------- /cifar/models/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/models/net.py -------------------------------------------------------------------------------- /cifar/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/models/resnet.py -------------------------------------------------------------------------------- /cifar/models/resnet_bcl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/models/resnet_bcl.py -------------------------------------------------------------------------------- /cifar/models/resnet_ncl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/models/resnet_ncl.py -------------------------------------------------------------------------------- /cifar/models/resnet_ride.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/models/resnet_ride.py -------------------------------------------------------------------------------- /cifar/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/train/train.py -------------------------------------------------------------------------------- /cifar/train/train_fn/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/train/train_fn/base.py -------------------------------------------------------------------------------- /cifar/train/train_fn/bcl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/train/train_fn/bcl.py -------------------------------------------------------------------------------- /cifar/train/train_fn/ncl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/train/train_fn/ncl.py -------------------------------------------------------------------------------- /cifar/train/train_fn/ride.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/train/train_fn/ride.py -------------------------------------------------------------------------------- /cifar/train/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/train/validate.py -------------------------------------------------------------------------------- /cifar/utils/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/utils/accuracy.py -------------------------------------------------------------------------------- /cifar/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/utils/common.py -------------------------------------------------------------------------------- /cifar/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/utils/config.py -------------------------------------------------------------------------------- /cifar/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/utils/logger.py -------------------------------------------------------------------------------- /cifar/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/cifar/utils/plot.py -------------------------------------------------------------------------------- /large_scale/aug/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/aug/cuda.py -------------------------------------------------------------------------------- /large_scale/aug/randaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/aug/randaug.py -------------------------------------------------------------------------------- /large_scale/aug/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/aug/transforms.py -------------------------------------------------------------------------------- /large_scale/config/iNaturalist/iNaturalist18_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/config/iNaturalist/iNaturalist18_val.txt -------------------------------------------------------------------------------- /large_scale/imagenet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/imagenet_train.py -------------------------------------------------------------------------------- /large_scale/imbalance_data/cmo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/imbalance_data/cmo.py -------------------------------------------------------------------------------- /large_scale/imbalance_data/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/imbalance_data/cutout.py -------------------------------------------------------------------------------- /large_scale/imbalance_data/cv_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/imbalance_data/cv_aug.py -------------------------------------------------------------------------------- /large_scale/imbalance_data/lt_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/imbalance_data/lt_data.py -------------------------------------------------------------------------------- /large_scale/inat18_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/inat18_train.py -------------------------------------------------------------------------------- /large_scale/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/losses.py -------------------------------------------------------------------------------- /large_scale/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/opts.py -------------------------------------------------------------------------------- /large_scale/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/test.py -------------------------------------------------------------------------------- /large_scale/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/train.py -------------------------------------------------------------------------------- /large_scale/training_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/training_functions.py -------------------------------------------------------------------------------- /large_scale/util/autoaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/util/autoaug.py -------------------------------------------------------------------------------- /large_scale/util/moco_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/util/moco_loader.py -------------------------------------------------------------------------------- /large_scale/util/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/util/randaugment.py -------------------------------------------------------------------------------- /large_scale/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sumyeongahn/CUDA_LTR/HEAD/large_scale/util/util.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | progress 2 | sacred 3 | Wand 4 | --------------------------------------------------------------------------------