├── lib ├── core │ └── __init__.py ├── utils │ └── __init__.py ├── data_transform │ └── __init__.py ├── loss │ ├── loss_impl │ │ ├── __init__.py │ │ └── cost_sensitive_cross_entropy_loss.py │ ├── utils.py │ ├── __init__.py │ └── loss_base.py ├── net │ └── __init__.py ├── config │ └── __init__.py ├── modules │ ├── __init__.py │ ├── pooling_ops.py │ └── classifier_ops.py ├── backbone │ └── __init__.py └── dataset │ └── __init__.py ├── data_parallel_train.sh ├── distributed_data_parallel_train.sh ├── main └── _init_paths.py ├── configs ├── iNat18 │ └── iNat18_baseline.yaml ├── test │ ├── data_parallel.yaml │ └── distributed_data_parallel.yaml ├── cao_cifar │ ├── baseline │ │ ├── cifar100_im50.yaml │ │ ├── cifar10_im100.yaml │ │ ├── cifar10_im50.yaml │ │ └── cifar100_im100.yaml │ ├── re_weighting │ │ ├── bsce │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ ├── cifar10_im100.yaml │ │ │ └── cifar100_im100.yaml │ │ ├── cdt │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ │ ├── focal │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ │ ├── cbce │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ └── cifar10_im100.yaml │ │ ├── csce │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar10_im100.yaml │ │ │ ├── cifar10_im50_square.yaml │ │ │ ├── cifar100_im100_square.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ ├── cifar10_im100_square.yaml │ │ │ └── cifar100_im50_square.yaml │ │ ├── seql │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ │ ├── ldam │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ │ ├── cels │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ │ ├── cbfocal │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ │ └── celas │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar10_im100.yaml │ ├── DiVE │ │ ├── cifar100_im100 │ │ │ ├── teacher.yaml │ │ │ └── student.yaml │ │ ├── cifar10_im100 │ │ │ ├── teacher.yaml │ │ │ └── student.yaml │ │ ├── cifar100_im50 │ │ │ ├── teacher.yaml │ │ │ └── student.yaml │ │ └── cifar10_im50 │ │ │ ├── teacher.yaml │ │ │ └── student.yaml │ ├── two_stage │ │ ├── drs │ │ │ ├── cam_based_sampling │ │ │ │ └── first_stage │ │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ │ └── cifar100_im100.yaml │ │ │ ├── balance │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ └── cifar100_im100.yaml │ │ │ ├── square │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ └── cifar10_im50.yaml │ │ │ └── progressive │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ └── cifar10_im50.yaml │ │ ├── drw │ │ │ ├── bsce │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ └── cifar100_im100.yaml │ │ │ ├── cdt │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ └── cifar10_im100.yaml │ │ │ ├── focal │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ └── cifar10_im100.yaml │ │ │ ├── cbce │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ └── cifar10_im100.yaml │ │ │ ├── csce │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ └── cifar100_im50.yaml │ │ │ ├── ldam │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ └── cifar10_im100.yaml │ │ │ ├── cels │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ └── cifar100_im100.yaml │ │ │ ├── cbfocal │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ └── cifar10_im100.yaml │ │ │ ├── ibloss │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar10_im100.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ └── cifar100_im50.yaml │ │ │ └── celas │ │ │ │ ├── cifar10_im50.yaml │ │ │ │ ├── cifar100_im100.yaml │ │ │ │ ├── cifar100_im50.yaml │ │ │ │ └── cifar10_im100.yaml │ │ └── classifier_balance │ │ │ ├── tau_norm │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar10_im100.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ └── cifar100_im50.yaml │ │ │ ├── cRT │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar10_im100.yaml │ │ │ ├── cifar100_im100.yaml │ │ │ └── cifar100_im50.yaml │ │ │ └── LWS │ │ │ ├── cifar10_im50.yaml │ │ │ ├── cifar10_im100.yaml │ │ │ ├── cifar100_im50.yaml │ │ │ └── cifar100_im100.yaml │ ├── mixup │ │ ├── input_mixup │ │ │ ├── cifar10_im50_im_alpha10.yaml │ │ │ ├── cifar100_im50_im_alpha10.yaml │ │ │ ├── cifar10_im100_im_alpha10.yaml │ │ │ └── cifar100_im100_im_alpha10.yaml │ │ ├── remix │ │ │ ├── cifar10_im50_remix_alpha10.yaml │ │ │ ├── cifar100_im100_remix_alpha10.yaml │ │ │ ├── cifar100_im50_remix_alpha10.yaml │ │ │ └── cifar10_im100_remix_alpha10.yaml │ │ └── manifold_mixup │ │ │ ├── cifar100_im50_mm_alpha10.yaml │ │ │ ├── cifar10_im100_mm_alpha10.yaml │ │ │ ├── cifar100_im100_mm_alpha10.yaml │ │ │ └── cifar10_im50_mm_alpha10.yaml │ └── re_sampling │ │ ├── bbn-style │ │ ├── cifar10_im50.yaml │ │ ├── cifar100_im50.yaml │ │ ├── cifar10_im100.yaml │ │ └── cifar100_im100.yaml │ │ ├── balance │ │ ├── cifar10_im50.yaml │ │ ├── cifar100_im100.yaml │ │ ├── cifar100_im50.yaml │ │ └── cifar10_im100.yaml │ │ ├── square │ │ ├── cifar10_im50.yaml │ │ ├── cifar100_im50.yaml │ │ ├── cifar10_im100.yaml │ │ └── cifar100_im100.yaml │ │ └── progressive │ │ ├── cifar100_im50.yaml │ │ ├── cifar10_im100.yaml │ │ ├── cifar10_im50.yaml │ │ └── cifar100_im100.yaml └── ImageNet_LT │ ├── imagenetlt_baseline.yaml │ └── combinations │ └── bag_of_tricks │ └── IM-DRSwithCAMbasedsampling-Finetune │ └── first_stage.yaml └── LICENSE /lib/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/data_transform/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/loss/loss_impl/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /lib/net/__init__.py: -------------------------------------------------------------------------------- 1 | from net.network import Network 2 | 3 | __all__ = [ 4 | 'Network', 5 | ] 6 | -------------------------------------------------------------------------------- /lib/config/__init__.py: -------------------------------------------------------------------------------- 1 | from config.default import _C as cfg 2 | from config.default import update_config 3 | 4 | __all__ = [ 5 | 'cfg', 'update_config' 6 | ] -------------------------------------------------------------------------------- /lib/modules/__init__.py: -------------------------------------------------------------------------------- 1 | from .classifier_ops import FCNorm, LWS 2 | from .pooling_ops import GAP, Identity 3 | 4 | __all__ = [ 5 | 'FCNorm', 'LWS', 'GAP', 'Identity' 6 | ] 7 | -------------------------------------------------------------------------------- /data_parallel_train.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | PYTHON=${PYTHON:-"python"} 4 | CONFIG=$1 5 | GPUS=$2 6 | 7 | CUDA_VISIBLE_DEVICES=$GPUS $PYTHON main/train.py --cfg $CONFIG ${@:3} 8 | -------------------------------------------------------------------------------- /lib/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | from .resnet import res50 2 | from .resnet_cifar import res32_cifar 3 | from .oltr_resnet import res10 4 | from .resnext import resnext50 5 | __all__ = [ 6 | 'res50', 'res32_cifar', 'res10', 'resnext50' 7 | ] 8 | -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | from .cao_cifar import IMBALANCECIFAR10, IMBALANCECIFAR100 2 | from .iNaturalist import iNaturalist 3 | from .cui_cifar import CIFAR 4 | 5 | __all__ = [ 6 | 'IMBALANCECIFAR10', 'IMBALANCECIFAR100', 'iNaturalist', 'CIFAR' 7 | ] 8 | -------------------------------------------------------------------------------- /lib/loss/utils.py: -------------------------------------------------------------------------------- 1 | import torch 2 | 3 | def get_one_hot(label, num_classes): 4 | batch_size = label.shape[0] 5 | onehot_label = torch.zeros((batch_size, num_classes)) 6 | onehot_label = onehot_label.scatter_(1, label.unsqueeze(1).detach().cpu(), 1) 7 | onehot_label = (onehot_label.type(torch.FloatTensor)).to(label.device) 8 | return onehot_label -------------------------------------------------------------------------------- /distributed_data_parallel_train.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | export NCCL_P2P_DISABLE=1 3 | # export NCCL_DEBUG=INFO 4 | export NCCL_IB_DISABLE=1 5 | export NCCL_SOCKET_IFNAME=enp97s0f0 6 | 7 | 8 | PYTHON=${PYTHON:-"python"} 9 | CONFIG=$1 10 | NUM_GPUS=$2 11 | GPUS=$3 12 | PORT=${PORT:-29500} 13 | 14 | 15 | CUDA_VISIBLE_DEVICES=$GPUS $PYTHON -m torch.distributed.launch --nproc_per_node=$NUM_GPUS --master_port=$PORT \ 16 | main/train.py --cfg $CONFIG ${@:4} 17 | -------------------------------------------------------------------------------- /main/_init_paths.py: -------------------------------------------------------------------------------- 1 | import sys 2 | import os 3 | from os import path as osp 4 | 5 | def add_path(path): 6 | if path not in sys.path: 7 | sys.path.insert(0, path) 8 | try: 9 | os.environ["PYTHONPATH"] = path + ":" + os.environ["PYTHONPATH"] 10 | except KeyError: 11 | os.environ["PYTHONPATH"] = path 12 | 13 | this_dir = osp.dirname(__file__) 14 | 15 | # Add lib to PYTHONPATH 16 | lib_path = osp.join(this_dir, '..', 'lib') 17 | add_path(lib_path) 18 | -------------------------------------------------------------------------------- /lib/modules/pooling_ops.py: -------------------------------------------------------------------------------- 1 | import torch 2 | import torch.nn as nn 3 | import torch.nn.functional as F 4 | 5 | 6 | class GAP(nn.Module): 7 | """Global Average pooling 8 | Widely used in ResNet, Inception, DenseNet, etc. 9 | """ 10 | 11 | def __init__(self): 12 | super(GAP, self).__init__() 13 | self.avgpool = nn.AdaptiveAvgPool2d((1, 1)) 14 | 15 | def forward(self, x): 16 | x = self.avgpool(x) 17 | # x = x.view(x.shape[0], -1) 18 | return x 19 | 20 | class Identity(nn.Module): 21 | def __init__(self): 22 | super(Identity, self).__init__() 23 | 24 | def forward(self, x): 25 | return x -------------------------------------------------------------------------------- /lib/loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .loss_base import CrossEntropy 2 | from .loss_impl.balanced_softmax_cross_entropy_loss import BalancedSoftmaxCE 3 | from .loss_impl.class_balanced_loss import ClassBalanceCE, ClassBalanceFocal 4 | from .loss_impl.class_dependent_temperatures_loss import CDT 5 | from .loss_impl.cost_sensitive_cross_entropy_loss import CostSensitiveCE 6 | from .loss_impl.cross_entropy_label_aware_smooth_loss import CrossEntropyLabelAwareSmooth 7 | from .loss_impl.cross_entropy_label_smooth_loss import CrossEntropyLabelSmooth 8 | from .loss_impl.equalization_loss import SEQL 9 | from .loss_impl.focal_loss import FocalLoss 10 | from .loss_impl.influence_balanced_loss import InfluenceBalancedLoss 11 | from .loss_impl.ldam_loss import LDAMLoss 12 | from .loss_impl.kld_loss import DiVEKLD 13 | 14 | __all__ = [ 15 | 'CrossEntropy', 'BalancedSoftmaxCE', 'ClassBalanceCE', 'ClassBalanceFocal', 16 | 'CDT', 'CostSensitiveCE', 'CrossEntropyLabelSmooth', 'CrossEntropyLabelAwareSmooth', 17 | 'SEQL', 'FocalLoss', 'InfluenceBalancedLoss', 'LDAMLoss', 'DiVEKLD' 18 | ] 19 | -------------------------------------------------------------------------------- /lib/modules/classifier_ops.py: -------------------------------------------------------------------------------- 1 | import torch 2 | import torch.nn as nn 3 | import torch.nn.functional as F 4 | 5 | # for LDAM Loss 6 | class FCNorm(nn.Module): 7 | def __init__(self, num_features, num_classes): 8 | super(FCNorm, self).__init__() 9 | self.weight = nn.Parameter(torch.FloatTensor(num_classes, num_features)) 10 | self.weight.data.uniform_(-1, 1).renorm_(2, 1, 1e-5).mul_(1e5) 11 | 12 | def forward(self, x): 13 | out = F.linear(F.normalize(x), F.normalize(self.weight)) 14 | return out 15 | 16 | 17 | class LWS(nn.Module): 18 | 19 | def __init__(self, num_features, num_classes, bias=True): 20 | super(LWS, self).__init__() 21 | self.fc = nn.Linear(num_features, num_classes, bias=bias) 22 | self.scales = nn.Parameter(torch.ones(num_classes)) 23 | for param_name, param in self.fc.named_parameters(): 24 | param.requires_grad = False 25 | 26 | def forward(self, x): 27 | x = self.fc(x) 28 | x *= self.scales 29 | return x 30 | -------------------------------------------------------------------------------- /configs/iNat18/iNat18_baseline.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/iNat18/' 3 | SHOW_STEP: 500 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (224, 224) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'iNaturalist' 11 | TRAIN_JSON: '/mnt/ramdisk/iNat/iNaturalist18_train.json' 12 | VALID_JSON: '/mnt/ramdisk/iNat/iNaturalist18_val.json' 13 | 14 | BACKBONE: 15 | TYPE: 'res50' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 100 30 | NUM_WORKERS: 8 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.2 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 1e-4 38 | LR_SCHEDULER: 39 | TYPE: 'multistep' 40 | LR_STEP: [60, 80] 41 | LR_FACTOR: 0.1 42 | TENSORBOARD: 43 | ENABLE: True 44 | SAMPLER: 45 | TYPE: 'default' 46 | 47 | DISTRIBUTED: True 48 | SYNCBN: True 49 | 50 | TEST: 51 | BATCH_SIZE: 512 52 | NUM_WORKERS: 8 53 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | MIT License 2 | 3 | Copyright (c) 2019 Megvii-Nanjing 4 | 5 | Permission is hereby granted, free of charge, to any person obtaining a copy 6 | of this software and associated documentation files (the "Software"), to deal 7 | in the Software without restriction, including without limitation the rights 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 | copies of the Software, and to permit persons to whom the Software is 10 | furnished to do so, subject to the following conditions: 11 | 12 | The above copyright notice and this permission notice shall be included in all 13 | copies or substantial portions of the Software. 14 | 15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 21 | SOFTWARE. 22 | -------------------------------------------------------------------------------- /configs/test/data_parallel.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar10_im50/baseline' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/baseline/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR100' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/baseline/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.01 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/baseline/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar10_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/baseline/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR100' 11 | IMBALANCECIFAR: 12 | RATIO: 0.01 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/test/distributed_data_parallel.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar10_im50/baseline' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/bsce/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'BalancedSoftmaxCE' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/bsce/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/bsce/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar100_im100/teacher.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive_teacher' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [120, 160] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar10_im100/teacher.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive_teacher' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [120, 160] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/bsce/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar100_im50/teacher.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive_teacher' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [120, 160] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar10_im50/teacher.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive_teacher' 2 | OUTPUT_DIR: './output/cifar10_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [120, 160] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cdt/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CDT' 26 | CDT: 27 | GAMMA: 0.2 28 | 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cdt/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CDT' 28 | CDT: 29 | GAMMA: 0.2 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cdt/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CDT' 28 | CDT: 29 | GAMMA: 0.2 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cdt/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CDT' 28 | CDT: 29 | GAMMA: 0.2 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/focal/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'FocalLoss' 26 | FocalLoss: 27 | GAMMA: 2.0 28 | 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/cam_based_sampling/first_stage/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'first_stage' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 160 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/cam_based_sampling/first_stage/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'first_stage' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 160 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/cam_based_sampling/first_stage/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'first_stage' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 160 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/input_mixup/cifar10_im50_im_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'input_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 400 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'mix_up' 33 | ALPHA: 1.0 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [320, 360] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbce/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'ClassBalanceCE' 26 | ClassBalanceCE: 27 | BETA: 0.9999 28 | 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CostSensitiveCE' 26 | CostSensitiveCE: 27 | GAMMA: 1.0 28 | 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/focal/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'FocalLoss' 27 | FocalLoss: 28 | GAMMA: 2.0 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/focal/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'FocalLoss' 27 | FocalLoss: 28 | GAMMA: 2.0 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/focal/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'FocalLoss' 27 | FocalLoss: 28 | GAMMA: 2.0 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/seql/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'seql' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'SEQL' 26 | SEQL: 27 | GAMMA: 0.9 28 | LAMBDA: 0.05 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/cam_based_sampling/first_stage/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'first_stage' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 160 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/input_mixup/cifar100_im50_im_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'input_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'mix_up' 34 | ALPHA: 1.0 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [320, 360] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/input_mixup/cifar10_im100_im_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'input_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'mix_up' 34 | ALPHA: 1.0 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [320, 360] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbce/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'ClassBalanceCE' 27 | ClassBalanceCE: 28 | BETA: 0.9 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/input_mixup/cifar100_im100_im_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'input_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'mix_up' 34 | ALPHA: 1.0 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [320, 360] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbce/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'ClassBalanceCE' 27 | ClassBalanceCE: 28 | BETA: 0.9 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbce/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'ClassBalanceCE' 27 | ClassBalanceCE: 28 | BETA: 0.9999 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar10_im50_square.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce_square' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CostSensitiveCE' 26 | CostSensitiveCE: 27 | GAMMA: 0.5 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/seql/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'seql' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'SEQL' 27 | SEQL: 28 | GAMMA: 0.9 29 | LAMBDA: 0.005 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/seql/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'seql' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR100' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'SEQL' 27 | SEQL: 28 | GAMMA: 0.9 29 | LAMBDA: 0.006 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/seql/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'seql' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'SEQL' 27 | SEQL: 28 | GAMMA: 0.9 29 | LAMBDA: 0.05 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar100_im100_square.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce_square' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR100' 11 | IMBALANCECIFAR: 12 | RATIO: 0.01 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CostSensitiveCE' 26 | CostSensitiveCE: 27 | GAMMA: 0.5 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/ldam/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FCNorm' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'LDAMLoss' 26 | LDAMLoss: 27 | SCALE: 30.0 28 | MAX_MARGIN: 0.5 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cels/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 26 | CrossEntropyLabelSmooth: 27 | EPSILON: 0.1 28 | 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar10_im100_square.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce_square' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/bbn-style/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bbn_style_sampling' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 64 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'bbn sampler' 47 | BBN_SAMPLER: 48 | TYPE: 'reverse' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbfocal/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'ClassBalanceFocal' 26 | ClassBalanceFocal: 27 | BETA: 0.9999 28 | GAMMA: 2.0 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cels/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR100' 11 | IMBALANCECIFAR: 12 | RATIO: 0.01 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 26 | CrossEntropyLabelSmooth: 27 | EPSILON: 0.1 28 | 29 | TRAIN: 30 | BATCH_SIZE: 128 31 | MAX_EPOCH: 200 32 | NUM_WORKERS: 16 33 | COMBINER: 34 | TYPE: 'default' 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.1 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 2e-4 40 | LR_SCHEDULER: 41 | TYPE: 'warmup' 42 | LR_STEP: [160, 180] 43 | LR_FACTOR: 0.01 44 | WARM_EPOCH: 5 45 | TENSORBOARD: 46 | ENABLE: True 47 | SAMPLER: 48 | TYPE: 'default' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cels/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 27 | CrossEntropyLabelSmooth: 28 | EPSILON: 0.1 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cels/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 27 | CrossEntropyLabelSmooth: 28 | EPSILON: 0.1 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/csce/cifar100_im50_square.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce_square' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 0.5 29 | 30 | TRAIN: 31 | BATCH_SIZE: 128 32 | MAX_EPOCH: 200 33 | NUM_WORKERS: 16 34 | COMBINER: 35 | TYPE: 'default' 36 | OPTIMIZER: 37 | TYPE: 'SGD' 38 | BASE_LR: 0.1 39 | MOMENTUM: 0.9 40 | WEIGHT_DECAY: 2e-4 41 | LR_SCHEDULER: 42 | TYPE: 'warmup' 43 | LR_STEP: [160, 180] 44 | LR_FACTOR: 0.01 45 | WARM_EPOCH: 5 46 | TENSORBOARD: 47 | ENABLE: True 48 | SAMPLER: 49 | TYPE: 'default' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/ldam/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FCNorm' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'LDAMLoss' 27 | LDAMLoss: 28 | SCALE: 30.0 29 | MAX_MARGIN: 0.5 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/ldam/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FCNorm' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'LDAMLoss' 27 | LDAMLoss: 28 | SCALE: 30.0 29 | MAX_MARGIN: 0.5 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/ldam/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FCNorm' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'LDAMLoss' 27 | LDAMLoss: 28 | SCALE: 30.0 29 | MAX_MARGIN: 0.5 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 200 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [160, 180] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/bbn-style/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bbn_style_sampling' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 64 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'bbn sampler' 48 | BBN_SAMPLER: 49 | TYPE: 'reverse' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/bbn-style/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bbn_style_sampling' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 64 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'bbn sampler' 48 | BBN_SAMPLER: 49 | TYPE: 'reverse' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/balance/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'class_balance_sampling' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'weighted sampler' 47 | WEIGHTED_SAMPLER: 48 | TYPE: 'balance' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/bbn-style/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bbn_style_sampling' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 64 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'bbn sampler' 48 | BBN_SAMPLER: 49 | TYPE: 'reverse' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/square/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'square_root_sampling' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'weighted sampler' 47 | WEIGHTED_SAMPLER: 48 | TYPE: 'square' 49 | 50 | DISTRIBUTED: False 51 | SYNCBN: True 52 | 53 | TRANSFORMS: 54 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 55 | TEST_TRANSFORMS: () 56 | 57 | TEST: 58 | BATCH_SIZE: 64 59 | NUM_WORKERS: 8 60 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/square/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'square_root_sampling' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'square' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/square/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'square_root_sampling' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'square' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/bsce/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | TWO_STAGE: 30 | DRW: True 31 | START_EPOCH: 160 32 | 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/bsce/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'BalancedSoftmaxCE' 26 | 27 | TRAIN: 28 | 29 | TWO_STAGE: 30 | DRW: True 31 | START_EPOCH: 160 32 | 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/remix/cifar10_im50_remix_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'remix_alpha10' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 400 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'remix' 33 | ALPHA: 1.0 34 | REMIX: 35 | KAPPA: 3.0 36 | TAU: 0.5 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [320, 360] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/balance/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'class_balance_sampling' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'balance' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/balance/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'class_balance_sampling' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'balance' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/balance/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'class_balance_sampling' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'balance' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/square/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'square_root_sampling' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'square' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbfocal/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'ClassBalanceFocal' 28 | ClassBalanceFocal: 29 | BETA: 0.9 30 | GAMMA: 1.0 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbfocal/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'ClassBalanceFocal' 28 | ClassBalanceFocal: 29 | BETA: 0.99 30 | GAMMA: 1.0 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/cbfocal/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'ClassBalanceFocal' 28 | ClassBalanceFocal: 29 | BETA: 0.9999 30 | GAMMA: 1.0 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/bsce/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRW: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'default' 53 | 54 | DISTRIBUTED: False 55 | SYNCBN: True 56 | 57 | TRANSFORMS: 58 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 59 | TEST_TRANSFORMS: () 60 | 61 | TEST: 62 | BATCH_SIZE: 64 63 | NUM_WORKERS: 8 64 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/bsce/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bsce-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'BalancedSoftmaxCE' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRW: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'default' 53 | 54 | DISTRIBUTED: False 55 | SYNCBN: True 56 | 57 | TRANSFORMS: 58 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 59 | TEST_TRANSFORMS: () 60 | 61 | TEST: 62 | BATCH_SIZE: 64 63 | NUM_WORKERS: 8 64 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/remix/cifar100_im100_remix_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'remix_alpha10' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'remix' 34 | ALPHA: 1.0 35 | REMIX: 36 | KAPPA: 3.0 37 | TAU: 0.5 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [320, 360] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/remix/cifar100_im50_remix_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'remix_alpha10' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'remix' 34 | ALPHA: 1.0 35 | REMIX: 36 | KAPPA: 3.0 37 | TAU: 0.5 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [320, 360] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/remix/cifar10_im100_remix_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'remix_alpha10' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'remix' 34 | ALPHA: 1.0 35 | REMIX: 36 | KAPPA: 3.0 37 | TAU: 0.5 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [320, 360] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cdt/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CDT' 26 | CDT: 27 | GAMMA: 0.2 28 | 29 | TRAIN: 30 | 31 | TWO_STAGE: 32 | DRW: True 33 | START_EPOCH: 160 34 | 35 | BATCH_SIZE: 128 36 | MAX_EPOCH: 200 37 | NUM_WORKERS: 16 38 | COMBINER: 39 | TYPE: 'default' 40 | OPTIMIZER: 41 | TYPE: 'SGD' 42 | BASE_LR: 0.1 43 | MOMENTUM: 0.9 44 | WEIGHT_DECAY: 2e-4 45 | LR_SCHEDULER: 46 | TYPE: 'warmup' 47 | LR_STEP: [160, 180] 48 | LR_FACTOR: 0.01 49 | WARM_EPOCH: 5 50 | TENSORBOARD: 51 | ENABLE: True 52 | SAMPLER: 53 | TYPE: 'default' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/progressive/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'progressively_balanced_sampling' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'progressive' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/progressive/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'progressively_balanced_sampling' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'progressive' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/progressive/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'progressively_balanced_sampling' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'weighted sampler' 47 | WEIGHTED_SAMPLER: 48 | TYPE: 'progressive' 49 | 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_sampling/progressive/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'progressively_balanced_sampling' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'weighted sampler' 48 | WEIGHTED_SAMPLER: 49 | TYPE: 'progressive' 50 | 51 | DISTRIBUTED: False 52 | SYNCBN: True 53 | 54 | TRANSFORMS: 55 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 56 | TEST_TRANSFORMS: () 57 | 58 | TEST: 59 | BATCH_SIZE: 64 60 | NUM_WORKERS: 8 61 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cdt/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CDT' 28 | CDT: 29 | GAMMA: 0.2 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cdt/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CDT' 28 | CDT: 29 | GAMMA: 0.2 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cdt/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cdt-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CDT' 28 | CDT: 29 | GAMMA: 0.2 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/focal/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal_drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'FocalLoss' 26 | FocalLoss: 27 | GAMMA: 2.0 28 | 29 | TRAIN: 30 | 31 | TWO_STAGE: 32 | DRW: True 33 | START_EPOCH: 160 34 | 35 | BATCH_SIZE: 128 36 | MAX_EPOCH: 200 37 | NUM_WORKERS: 16 38 | COMBINER: 39 | TYPE: 'default' 40 | OPTIMIZER: 41 | TYPE: 'SGD' 42 | BASE_LR: 0.1 43 | MOMENTUM: 0.9 44 | WEIGHT_DECAY: 2e-4 45 | LR_SCHEDULER: 46 | TYPE: 'warmup' 47 | LR_STEP: [160, 180] 48 | LR_FACTOR: 0.01 49 | WARM_EPOCH: 5 50 | TENSORBOARD: 51 | ENABLE: True 52 | SAMPLER: 53 | TYPE: 'default' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /lib/loss/loss_base.py: -------------------------------------------------------------------------------- 1 | import torch.nn as nn 2 | from torch.nn import functional as F 3 | 4 | class CrossEntropy(nn.Module): 5 | def __init__(self, para_dict=None): 6 | super(CrossEntropy, self).__init__() 7 | self.para_dict = para_dict 8 | self.num_classes = self.para_dict["num_classes"] 9 | self.num_class_list = self.para_dict['num_class_list'] 10 | self.device = self.para_dict['device'] 11 | 12 | self.weight_list = None 13 | 14 | #settings of defferred re-balancing by re-weighting (DRW) 15 | self.drw = self.para_dict['cfg'].TRAIN.TWO_STAGE.DRW 16 | self.drw_start_epoch = self.para_dict['cfg'].TRAIN.TWO_STAGE.START_EPOCH #start from 1 17 | 18 | def forward(self, inputs, targets, **kwargs): 19 | """ 20 | Args: 21 | inputs: prediction matrix (before softmax) with shape (batch_size, num_classes) 22 | targets: ground truth labels with shape (batch_size) 23 | """ 24 | loss = F.cross_entropy(inputs, targets, weight=self.weight_list) 25 | return loss 26 | 27 | def update(self, epoch): 28 | """ 29 | Adopt cost-sensitive cross-entropy as the default 30 | Args: 31 | epoch: int. starting from 1. 32 | """ 33 | pass -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/manifold_mixup/cifar100_im50_mm_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'manifold_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'manifold_mix_up' 34 | ALPHA: 1.0 35 | MANIFOLD_MIX_UP: 36 | LOCATION: "pool" 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [320, 360] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/manifold_mixup/cifar10_im100_mm_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'manifold_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'manifold_mix_up' 34 | ALPHA: 1.0 35 | MANIFOLD_MIX_UP: 36 | LOCATION: "pool" 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [320, 360] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/manifold_mixup/cifar100_im100_mm_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'manifold_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 400 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'manifold_mix_up' 34 | ALPHA: 1.0 35 | MANIFOLD_MIX_UP: 36 | LOCATION: "pool" 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 2e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [320, 360] 45 | LR_FACTOR: 0.01 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'default' 51 | 52 | DISTRIBUTED: False 53 | SYNCBN: True 54 | 55 | TRANSFORMS: 56 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 57 | TEST_TRANSFORMS: () 58 | 59 | TEST: 60 | BATCH_SIZE: 64 61 | NUM_WORKERS: 8 62 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/mixup/manifold_mixup/cifar10_im50_mm_alpha10.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'manifold_mixup_alpha10' 2 | OUTPUT_DIR: './output/cifar10_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 400 30 | NUM_WORKERS: 16 31 | 32 | COMBINER: 33 | TYPE: 'manifold_mix_up' 34 | ALPHA: 1.0 35 | MANIFOLD_MIX_UP: 36 | LOCATION: "pool" 37 | 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [320, 360] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | SYNCBN: True 55 | 56 | TRANSFORMS: 57 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 58 | TEST_TRANSFORMS: () 59 | 60 | TEST: 61 | BATCH_SIZE: 64 62 | NUM_WORKERS: 8 63 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/tau_norm/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 128 29 | MAX_EPOCH: 200 30 | NUM_WORKERS: 16 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.1 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 2e-4 38 | LR_SCHEDULER: 39 | TYPE: 'warmup' 40 | LR_STEP: [160, 180] 41 | LR_FACTOR: 0.01 42 | WARM_EPOCH: 5 43 | TENSORBOARD: 44 | ENABLE: True 45 | SAMPLER: 46 | TYPE: 'default' 47 | 48 | DISTRIBUTED: False 49 | SYNCBN: True 50 | 51 | TRANSFORMS: 52 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 53 | TEST_TRANSFORMS: () 54 | 55 | TEST: 56 | BATCH_SIZE: 64 57 | NUM_WORKERS: 8 58 | MODEL_FILE: './output/cifar10_im50/baseline/models/best_model.pth' 59 | TAU_NORM: 60 | USE_TAU_NORM: True 61 | TAU: 1.0 -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbce/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'ClassBalanceCE' 26 | ClassBalanceCE: 27 | BETA: 0.9999 28 | 29 | TRAIN: 30 | 31 | TWO_STAGE: 32 | DRW: True 33 | START_EPOCH: 160 34 | 35 | BATCH_SIZE: 128 36 | MAX_EPOCH: 200 37 | NUM_WORKERS: 16 38 | COMBINER: 39 | TYPE: 'default' 40 | OPTIMIZER: 41 | TYPE: 'SGD' 42 | BASE_LR: 0.1 43 | MOMENTUM: 0.9 44 | WEIGHT_DECAY: 2e-4 45 | LR_SCHEDULER: 46 | TYPE: 'warmup' 47 | LR_STEP: [160, 180] 48 | LR_FACTOR: 0.01 49 | WARM_EPOCH: 5 50 | TENSORBOARD: 51 | ENABLE: True 52 | SAMPLER: 53 | TYPE: 'default' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/focal/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'FocalLoss' 27 | FocalLoss: 28 | GAMMA: 2.0 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/focal/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal_drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'FocalLoss' 27 | FocalLoss: 28 | GAMMA: 2.0 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/focal/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'focal-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'FocalLoss' 27 | FocalLoss: 28 | GAMMA: 2.0 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/tau_norm/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: './output/cifar10_im100/baseline/models/best_model.pth' 60 | TAU_NORM: 61 | USE_TAU_NORM: True 62 | TAU: 1.0 -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbce/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'ClassBalanceCE' 27 | ClassBalanceCE: 28 | BETA: 0.9 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/csce/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CostSensitiveCE' 26 | CostSensitiveCE: 27 | GAMMA: 1.0 28 | 29 | TRAIN: 30 | 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/tau_norm/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: './output/cifar100_im100/baseline/models/best_model.pth' 60 | TAU_NORM: 61 | USE_TAU_NORM: True 62 | TAU: 1.0 -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/tau_norm/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | BATCH_SIZE: 128 30 | MAX_EPOCH: 200 31 | NUM_WORKERS: 16 32 | COMBINER: 33 | TYPE: 'default' 34 | OPTIMIZER: 35 | TYPE: 'SGD' 36 | BASE_LR: 0.1 37 | MOMENTUM: 0.9 38 | WEIGHT_DECAY: 2e-4 39 | LR_SCHEDULER: 40 | TYPE: 'warmup' 41 | LR_STEP: [160, 180] 42 | LR_FACTOR: 0.01 43 | WARM_EPOCH: 5 44 | TENSORBOARD: 45 | ENABLE: True 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: True 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 54 | TEST_TRANSFORMS: () 55 | 56 | TEST: 57 | BATCH_SIZE: 64 58 | NUM_WORKERS: 8 59 | MODEL_FILE: './output/cifar100_im50/baseline/models/best_model.pth' 60 | TAU_NORM: 61 | USE_TAU_NORM: True 62 | TAU: 1.0 -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbce/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'ClassBalanceCE' 27 | ClassBalanceCE: 28 | BETA: 0.9 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbce/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbce-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'ClassBalanceCE' 27 | ClassBalanceCE: 28 | BETA: 0.9999 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/csce/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/csce/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/csce/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'csce-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CostSensitiveCE' 27 | CostSensitiveCE: 28 | GAMMA: 1.0 29 | 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ldam/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FCNorm' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'LDAMLoss' 26 | LDAMLoss: 27 | SCALE: 30.0 28 | MAX_MARGIN: 0.5 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/balance/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbs-drs' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | 29 | TWO_STAGE: 30 | DRS: True 31 | START_EPOCH: 160 32 | 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'weighted sampler' 52 | WEIGHTED_SAMPLER: 53 | TYPE: 'balance' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cels/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 26 | CrossEntropyLabelSmooth: 27 | EPSILON: 0.1 28 | 29 | TRAIN: 30 | 31 | TWO_STAGE: 32 | DRW: True 33 | START_EPOCH: 160 34 | 35 | BATCH_SIZE: 128 36 | MAX_EPOCH: 200 37 | NUM_WORKERS: 16 38 | COMBINER: 39 | TYPE: 'default' 40 | OPTIMIZER: 41 | TYPE: 'SGD' 42 | BASE_LR: 0.1 43 | MOMENTUM: 0.9 44 | WEIGHT_DECAY: 2e-4 45 | LR_SCHEDULER: 46 | TYPE: 'warmup' 47 | LR_STEP: [160, 180] 48 | LR_FACTOR: 0.01 49 | WARM_EPOCH: 5 50 | TENSORBOARD: 51 | ENABLE: True 52 | SAMPLER: 53 | TYPE: 'default' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/balance/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbs-drs' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'balance' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/balance/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbs-drs' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'balance' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/square/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'srs-drs' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'square' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/square/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'srs-drs' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'square' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/square/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'srs-drs' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'square' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/square/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'srs-drs' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | 29 | TWO_STAGE: 30 | DRS: True 31 | START_EPOCH: 160 32 | 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'square' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbfocal/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'ClassBalanceFocal' 26 | ClassBalanceFocal: 27 | BETA: 0.9999 28 | GAMMA: 2.0 29 | TRAIN: 30 | 31 | TWO_STAGE: 32 | DRW: True 33 | START_EPOCH: 160 34 | 35 | BATCH_SIZE: 128 36 | MAX_EPOCH: 200 37 | NUM_WORKERS: 16 38 | COMBINER: 39 | TYPE: 'default' 40 | OPTIMIZER: 41 | TYPE: 'SGD' 42 | BASE_LR: 0.1 43 | MOMENTUM: 0.9 44 | WEIGHT_DECAY: 2e-4 45 | LR_SCHEDULER: 46 | TYPE: 'warmup' 47 | LR_STEP: [160, 180] 48 | LR_FACTOR: 0.01 49 | WARM_EPOCH: 5 50 | TENSORBOARD: 51 | ENABLE: True 52 | SAMPLER: 53 | TYPE: 'default' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cels/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 27 | CrossEntropyLabelSmooth: 28 | EPSILON: 0.1 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cels/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 27 | CrossEntropyLabelSmooth: 28 | EPSILON: 0.1 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ibloss/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ibloss-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'InfluenceBalancedLoss' 26 | InfluenceBalancedLoss: 27 | ALPHA: 1000. 28 | 29 | TRAIN: 30 | 31 | TWO_STAGE: 32 | DRW: True 33 | START_EPOCH: 100 34 | 35 | BATCH_SIZE: 128 36 | MAX_EPOCH: 200 37 | NUM_WORKERS: 16 38 | COMBINER: 39 | TYPE: 'default' 40 | OPTIMIZER: 41 | TYPE: 'SGD' 42 | BASE_LR: 0.1 43 | MOMENTUM: 0.9 44 | WEIGHT_DECAY: 2e-4 45 | LR_SCHEDULER: 46 | TYPE: 'warmup' 47 | LR_STEP: [100, 160, 180] 48 | LR_FACTOR: 0.01 49 | WARM_EPOCH: 5 50 | TENSORBOARD: 51 | ENABLE: True 52 | SAMPLER: 53 | TYPE: 'default' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ldam/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FCNorm' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'LDAMLoss' 27 | LDAMLoss: 28 | SCALE: 30.0 29 | MAX_MARGIN: 0.5 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ldam/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FCNorm' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'LDAMLoss' 27 | LDAMLoss: 28 | SCALE: 30.0 29 | MAX_MARGIN: 0.5 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ldam/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ldam-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FCNorm' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'LDAMLoss' 27 | LDAMLoss: 28 | SCALE: 30.0 29 | MAX_MARGIN: 0.5 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/ImageNet_LT/imagenetlt_baseline.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'baseline' 2 | OUTPUT_DIR: './output/ImageNet_LT/' 3 | SHOW_STEP: 200 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (224, 224) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'iNaturalist' 11 | TRAIN_JSON: '/mnt/data3/zhangys/programs/BagofTricks-LT/ImageNet_LT_train.json' 12 | VALID_JSON: '/mnt/data3/zhangys/programs/BagofTricks-LT/ImageNet_LT_val.json' 13 | 14 | BACKBONE: 15 | TYPE: 'res10' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 512 29 | MAX_EPOCH: 100 30 | NUM_WORKERS: 8 31 | COMBINER: 32 | TYPE: 'default' 33 | OPTIMIZER: 34 | TYPE: 'SGD' 35 | BASE_LR: 0.2 36 | MOMENTUM: 0.9 37 | WEIGHT_DECAY: 1e-4 38 | LR_SCHEDULER: 39 | TYPE: 'multistep' 40 | LR_STEP: [60, 80] 41 | LR_FACTOR: 0.1 42 | TENSORBOARD: 43 | ENABLE: True 44 | SAMPLER: 45 | TYPE: 'default' 46 | 47 | DISTRIBUTED: False 48 | SYNCBN: False 49 | 50 | TRANSFORMS: 51 | TRAIN_TRANSFORMS: ("random_resized_crop", "random_horizontal_flip", "color_jitter") 52 | TEST_TRANSFORMS: ("shorter_resize_for_crop", "center_crop") 53 | 54 | TEST: 55 | BATCH_SIZE: 512 56 | NUM_WORKERS: 8 57 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/celas/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 27 | CrossEntropyLabelAwareSmooth: 28 | SMOOTH_HEAD: 0.4 29 | SMOOTH_TAIL: 0.1 30 | SHAPE: 'concave' 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | APEX: False 55 | SYNCBN: True 56 | 57 | TRANSFORMS: 58 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 59 | TEST_TRANSFORMS: () 60 | 61 | TEST: 62 | BATCH_SIZE: 64 63 | NUM_WORKERS: 8 64 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/balance/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbs-drs' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'balance' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cels/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cels-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelSmooth' 27 | CrossEntropyLabelSmooth: 28 | EPSILON: 0.1 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 160 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ibloss/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ibloss-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'InfluenceBalancedLoss' 27 | InfluenceBalancedLoss: 28 | ALPHA: 1000. 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 100 35 | 36 | BATCH_SIZE: 128 37 | MAX_EPOCH: 200 38 | NUM_WORKERS: 16 39 | COMBINER: 40 | TYPE: 'default' 41 | OPTIMIZER: 42 | TYPE: 'SGD' 43 | BASE_LR: 0.1 44 | MOMENTUM: 0.9 45 | WEIGHT_DECAY: 2e-4 46 | LR_SCHEDULER: 47 | TYPE: 'warmup' 48 | LR_STEP: [100, 160, 180] 49 | LR_FACTOR: 0.01 50 | WARM_EPOCH: 5 51 | TENSORBOARD: 52 | ENABLE: True 53 | SAMPLER: 54 | TYPE: 'default' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/celas/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 27 | CrossEntropyLabelAwareSmooth: 28 | SMOOTH_HEAD: 0.4 29 | SMOOTH_TAIL: 0.1 30 | SHAPE: 'concave' 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'default' 52 | 53 | DISTRIBUTED: False 54 | APEX: False 55 | SYNCBN: True 56 | 57 | TRANSFORMS: 58 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 59 | TEST_TRANSFORMS: () 60 | 61 | TEST: 62 | BATCH_SIZE: 64 63 | NUM_WORKERS: 8 64 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/progressive/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'pbs-drs' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'progressive' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/progressive/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'pbs-drs' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'progressive' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/progressive/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'pbs-drs' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropy' 27 | 28 | TRAIN: 29 | 30 | TWO_STAGE: 31 | DRS: True 32 | START_EPOCH: 160 33 | 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'progressive' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drs/progressive/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'pbs-drs' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | 29 | TWO_STAGE: 30 | DRS: True 31 | START_EPOCH: 160 32 | 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 200 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 2e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [160, 180] 46 | LR_FACTOR: 0.01 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'weighted sampler' 52 | WEIGHTED_SAMPLER: 53 | TYPE: 'progressive' 54 | 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/celas/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 28 | CrossEntropyLabelAwareSmooth: 29 | SMOOTH_HEAD: 0.4 30 | SMOOTH_TAIL: 0.1 31 | SHAPE: 'concave' 32 | 33 | TRAIN: 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'default' 53 | 54 | DISTRIBUTED: False 55 | APEX: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/re_weighting/celas/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 28 | CrossEntropyLabelAwareSmooth: 29 | SMOOTH_HEAD: 0.4 30 | SMOOTH_TAIL: 0.1 31 | SHAPE: 'concave' 32 | 33 | TRAIN: 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 200 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 2e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [160, 180] 47 | LR_FACTOR: 0.01 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'default' 53 | 54 | DISTRIBUTED: False 55 | APEX: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ibloss/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ibloss-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'InfluenceBalancedLoss' 27 | InfluenceBalancedLoss: 28 | ALPHA: 1000. 29 | 30 | TRAIN: 31 | 32 | TWO_STAGE: 33 | DRW: True 34 | START_EPOCH: 100 35 | 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [100, 160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbfocal/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'ClassBalanceFocal' 28 | ClassBalanceFocal: 29 | BETA: 0.9 30 | GAMMA: 1.0 31 | 32 | TRAIN: 33 | 34 | TWO_STAGE: 35 | DRW: True 36 | START_EPOCH: 160 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbfocal/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'ClassBalanceFocal' 28 | ClassBalanceFocal: 29 | BETA: 0.99 30 | GAMMA: 1.0 31 | 32 | TRAIN: 33 | 34 | TWO_STAGE: 35 | DRW: True 36 | START_EPOCH: 160 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/cbfocal/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'cbfocal-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | 26 | LOSS: 27 | LOSS_TYPE: 'ClassBalanceFocal' 28 | ClassBalanceFocal: 29 | BETA: 0.9999 30 | GAMMA: 1.0 31 | 32 | TRAIN: 33 | 34 | TWO_STAGE: 35 | DRW: True 36 | START_EPOCH: 160 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/ibloss/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'ibloss-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'InfluenceBalancedLoss' 27 | InfluenceBalancedLoss: 28 | ALPHA: 1000. 29 | 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 100 36 | 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [100, 160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/celas/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas-drw' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 26 | CrossEntropyLabelAwareSmooth: 27 | SMOOTH_HEAD: 0.4 28 | SMOOTH_TAIL: 0.1 29 | SHAPE: 'concave' 30 | 31 | TRAIN: 32 | 33 | TWO_STAGE: 34 | DRW: True 35 | START_EPOCH: 160 36 | 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'default' 42 | OPTIMIZER: 43 | TYPE: 'SGD' 44 | BASE_LR: 0.1 45 | MOMENTUM: 0.9 46 | WEIGHT_DECAY: 2e-4 47 | LR_SCHEDULER: 48 | TYPE: 'warmup' 49 | LR_STEP: [160, 180] 50 | LR_FACTOR: 0.01 51 | WARM_EPOCH: 5 52 | TENSORBOARD: 53 | ENABLE: True 54 | SAMPLER: 55 | TYPE: 'default' 56 | 57 | DISTRIBUTED: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/cRT/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'crt_classifier_balance' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | NETWORK: 18 | PRETRAINED: True 19 | PRETRAINED_MODEL: './output/cifar10_im50/baseline/models/epoch_200.pth' 20 | 21 | MODULE: 22 | TYPE: 'GAP' 23 | 24 | CLASSIFIER: 25 | TYPE: 'cRT' 26 | BIAS: True 27 | 28 | LOSS: 29 | LOSS_TYPE: 'CrossEntropy' 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 30 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 5e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [10, 20] 45 | LR_FACTOR: 0.1 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'weighted sampler' 51 | WEIGHTED_SAMPLER: 52 | TYPE: 'balance' 53 | 54 | DISTRIBUTED: False 55 | SYNCBN: True 56 | 57 | TRANSFORMS: 58 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 59 | TEST_TRANSFORMS: () 60 | 61 | TEST: 62 | BATCH_SIZE: 64 63 | NUM_WORKERS: 8 64 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/LWS/cifar10_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'lws_classifier_balance' 2 | OUTPUT_DIR: './output/cifar10_im50/' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR10' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | NETWORK: 18 | PRETRAINED: True 19 | PRETRAINED_MODEL: './output/cifar10_im50/baseline/models/epoch_200.pth' 20 | 21 | 22 | MODULE: 23 | TYPE: 'GAP' 24 | 25 | CLASSIFIER: 26 | TYPE: 'LWS' 27 | BIAS: True 28 | 29 | LOSS: 30 | LOSS_TYPE: 'CrossEntropy' 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 30 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 5e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [10, 20] 46 | LR_FACTOR: 0.1 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'weighted sampler' 52 | WEIGHTED_SAMPLER: 53 | TYPE: 'balance' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/celas/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas-drw' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 27 | CrossEntropyLabelAwareSmooth: 28 | SMOOTH_HEAD: 0.4 29 | SMOOTH_TAIL: 0.1 30 | SHAPE: 'concave' 31 | 32 | TRAIN: 33 | 34 | TWO_STAGE: 35 | DRW: True 36 | START_EPOCH: 160 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/celas/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas-drw' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 27 | CrossEntropyLabelAwareSmooth: 28 | SMOOTH_HEAD: 0.4 29 | SMOOTH_TAIL: 0.1 30 | SHAPE: 'concave' 31 | 32 | TRAIN: 33 | 34 | TWO_STAGE: 35 | DRW: True 36 | START_EPOCH: 160 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/drw/celas/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'celas-drw' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'CrossEntropyLabelAwareSmooth' 27 | CrossEntropyLabelAwareSmooth: 28 | SMOOTH_HEAD: 0.4 29 | SMOOTH_TAIL: 0.1 30 | SHAPE: 'concave' 31 | 32 | TRAIN: 33 | 34 | TWO_STAGE: 35 | DRW: True 36 | START_EPOCH: 160 37 | 38 | BATCH_SIZE: 128 39 | MAX_EPOCH: 200 40 | NUM_WORKERS: 16 41 | COMBINER: 42 | TYPE: 'default' 43 | OPTIMIZER: 44 | TYPE: 'SGD' 45 | BASE_LR: 0.1 46 | MOMENTUM: 0.9 47 | WEIGHT_DECAY: 2e-4 48 | LR_SCHEDULER: 49 | TYPE: 'warmup' 50 | LR_STEP: [160, 180] 51 | LR_FACTOR: 0.01 52 | WARM_EPOCH: 5 53 | TENSORBOARD: 54 | ENABLE: True 55 | SAMPLER: 56 | TYPE: 'default' 57 | 58 | DISTRIBUTED: False 59 | SYNCBN: True 60 | 61 | TRANSFORMS: 62 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 63 | TEST_TRANSFORMS: () 64 | 65 | TEST: 66 | BATCH_SIZE: 64 67 | NUM_WORKERS: 8 68 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /lib/loss/loss_impl/cost_sensitive_cross_entropy_loss.py: -------------------------------------------------------------------------------- 1 | import torch 2 | import numpy as np 3 | 4 | from ..loss_base import CrossEntropy 5 | 6 | class CostSensitiveCE(CrossEntropy): 7 | r""" 8 | Equation: Loss(z, c) = - (\frac{N_min}{N_c})^\gamma * CrossEntropy(z, c), 9 | where gamma is a hyper-parameter to control the weights, 10 | N_min is the number of images in the smallest class, 11 | and N_c is the number of images in the class c. 12 | 13 | The representative re-weighting methods, which assigns class-dependent weights to the loss function 14 | 15 | Args: 16 | gamma (float or double): to control the loss weights: (N_min/N_i)^gamma 17 | """ 18 | def __init__(self, para_dict=None): 19 | super(CostSensitiveCE, self).__init__(para_dict) 20 | gamma = self.para_dict['cfg'].LOSS.CostSensitiveCE.GAMMA 21 | self.csce_weight = torch.FloatTensor(np.array([(min(self.num_class_list) / N)**gamma for N in self.num_class_list])).to(self.device) 22 | 23 | def update(self, epoch): 24 | """ 25 | Args: 26 | epoch: int. starting from 1. 27 | """ 28 | if not self.drw: 29 | self.weight_list = self.csce_weight 30 | else: 31 | start = (epoch-1) // self.drw_start_epoch 32 | if start: 33 | self.weight_list = self.csce_weight 34 | -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/cRT/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'crt_classifier_balance' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | NETWORK: 19 | PRETRAINED: True 20 | PRETRAINED_MODEL: './output/cifar10_im100/baseline/models/epoch_200.pth' 21 | 22 | MODULE: 23 | TYPE: 'GAP' 24 | 25 | CLASSIFIER: 26 | TYPE: 'cRT' 27 | BIAS: True 28 | 29 | LOSS: 30 | LOSS_TYPE: 'CrossEntropy' 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 30 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 5e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [10, 20] 46 | LR_FACTOR: 0.1 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'weighted sampler' 52 | WEIGHTED_SAMPLER: 53 | TYPE: 'balance' 54 | 55 | DISTRIBUTED: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/LWS/cifar10_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'lws_classifier_balance' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | NETWORK: 19 | PRETRAINED: True 20 | PRETRAINED_MODEL: './output/cifar10_im100/baseline/models/epoch_200.pth' 21 | 22 | 23 | MODULE: 24 | TYPE: 'GAP' 25 | 26 | CLASSIFIER: 27 | TYPE: 'LWS' 28 | BIAS: True 29 | 30 | LOSS: 31 | LOSS_TYPE: 'CrossEntropy' 32 | 33 | TRAIN: 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 30 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 5e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [10, 20] 47 | LR_FACTOR: 0.1 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'balance' 55 | 56 | DISTRIBUTED: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/LWS/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'lws_classifier_balance' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'IMBALANCECIFAR100' 11 | IMBALANCECIFAR: 12 | RATIO: 0.02 13 | 14 | BACKBONE: 15 | TYPE: 'res32_cifar' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | NETWORK: 21 | PRETRAINED: True 22 | PRETRAINED_MODEL: './output/cifar100_im50/baseline/models/epoch_200.pth' 23 | 24 | CLASSIFIER: 25 | TYPE: 'LWS' 26 | BIAS: True 27 | 28 | LOSS: 29 | LOSS_TYPE: 'CrossEntropy' 30 | 31 | TRAIN: 32 | BATCH_SIZE: 128 33 | MAX_EPOCH: 30 34 | NUM_WORKERS: 16 35 | COMBINER: 36 | TYPE: 'default' 37 | OPTIMIZER: 38 | TYPE: 'SGD' 39 | BASE_LR: 0.1 40 | MOMENTUM: 0.9 41 | WEIGHT_DECAY: 5e-4 42 | LR_SCHEDULER: 43 | TYPE: 'warmup' 44 | LR_STEP: [10, 20] 45 | LR_FACTOR: 0.1 46 | WARM_EPOCH: 5 47 | TENSORBOARD: 48 | ENABLE: True 49 | SAMPLER: 50 | TYPE: 'weighted sampler' 51 | WEIGHTED_SAMPLER: 52 | TYPE: 'balance' 53 | 54 | DISTRIBUTED: False 55 | APEX: False 56 | SYNCBN: True 57 | 58 | TRANSFORMS: 59 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 60 | TEST_TRANSFORMS: () 61 | 62 | TEST: 63 | BATCH_SIZE: 64 64 | NUM_WORKERS: 8 65 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/cRT/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'crt_classifier_balance' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | NETWORK: 19 | PRETRAINED: True 20 | PRETRAINED_MODEL: './output/cifar100_im100/baseline/models/epoch_200.pth' 21 | 22 | MODULE: 23 | TYPE: 'GAP' 24 | 25 | CLASSIFIER: 26 | TYPE: 'cRT' 27 | BIAS: True 28 | 29 | LOSS: 30 | LOSS_TYPE: 'CrossEntropy' 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 30 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 5e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [10, 20] 46 | LR_FACTOR: 0.1 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'weighted sampler' 52 | WEIGHTED_SAMPLER: 53 | TYPE: 'balance' 54 | 55 | DISTRIBUTED: False 56 | APEX: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/cRT/cifar100_im50.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'crt_classifier_balance' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | NETWORK: 19 | PRETRAINED: True 20 | PRETRAINED_MODEL: './output/cifar100_im50/baseline/models/epoch_200.pth' 21 | 22 | MODULE: 23 | TYPE: 'GAP' 24 | 25 | CLASSIFIER: 26 | TYPE: 'cRT' 27 | BIAS: True 28 | 29 | LOSS: 30 | LOSS_TYPE: 'CrossEntropy' 31 | 32 | TRAIN: 33 | BATCH_SIZE: 128 34 | MAX_EPOCH: 30 35 | NUM_WORKERS: 16 36 | COMBINER: 37 | TYPE: 'default' 38 | OPTIMIZER: 39 | TYPE: 'SGD' 40 | BASE_LR: 0.1 41 | MOMENTUM: 0.9 42 | WEIGHT_DECAY: 5e-4 43 | LR_SCHEDULER: 44 | TYPE: 'warmup' 45 | LR_STEP: [10, 20] 46 | LR_FACTOR: 0.1 47 | WARM_EPOCH: 5 48 | TENSORBOARD: 49 | ENABLE: True 50 | SAMPLER: 51 | TYPE: 'weighted sampler' 52 | WEIGHTED_SAMPLER: 53 | TYPE: 'balance' 54 | 55 | DISTRIBUTED: False 56 | APEX: False 57 | SYNCBN: True 58 | 59 | TRANSFORMS: 60 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 61 | TEST_TRANSFORMS: () 62 | 63 | TEST: 64 | BATCH_SIZE: 64 65 | NUM_WORKERS: 8 66 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/ImageNet_LT/combinations/bag_of_tricks/IM-DRSwithCAMbasedsampling-Finetune/first_stage.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'bagoftricks-first-stage' 2 | OUTPUT_DIR: './output/imagenet-lt/' 3 | SHOW_STEP: 200 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (224, 224) 7 | COLOR_SPACE: 'RGB' 8 | 9 | DATASET: 10 | DATASET: 'iNaturalist' 11 | TRAIN_JSON: '/mnt/data3/zhangys/programs/BagofTricks-LT/ImageNet_LT_train.json' 12 | VALID_JSON: '/mnt/data3/zhangys/programs/BagofTricks-LT/ImageNet_LT_val.json' 13 | 14 | BACKBONE: 15 | TYPE: 'res10' 16 | 17 | MODULE: 18 | TYPE: 'GAP' 19 | 20 | CLASSIFIER: 21 | TYPE: 'FC' 22 | BIAS: True 23 | 24 | LOSS: 25 | LOSS_TYPE: 'CrossEntropy' 26 | 27 | TRAIN: 28 | BATCH_SIZE: 512 29 | MAX_EPOCH: 160 30 | NUM_WORKERS: 8 31 | COMBINER: 32 | TYPE: 'mix_up' 33 | ALPHA: 1.0 34 | 35 | OPTIMIZER: 36 | TYPE: 'SGD' 37 | BASE_LR: 0.2 38 | MOMENTUM: 0.9 39 | WEIGHT_DECAY: 1e-4 40 | LR_SCHEDULER: 41 | TYPE: 'multistep' 42 | LR_STEP: [120, 160] 43 | LR_FACTOR: 0.1 44 | TENSORBOARD: 45 | ENABLE: False 46 | SAMPLER: 47 | TYPE: 'default' 48 | 49 | DISTRIBUTED: False 50 | SYNCBN: False 51 | 52 | TRANSFORMS: 53 | TRAIN_TRANSFORMS: ("random_resized_crop", "random_horizontal_flip", "color_jitter") 54 | TEST_TRANSFORMS: ("shorter_resize_for_crop", "center_crop") 55 | 56 | TEST: 57 | BATCH_SIZE: 512 58 | NUM_WORKERS: 8 59 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/two_stage/classifier_balance/LWS/cifar100_im100.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'lws_classifier_balance' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | NETWORK: 19 | PRETRAINED: True 20 | PRETRAINED_MODEL: './output/cifar100_im100/baseline/models/epoch_200.pth' 21 | 22 | 23 | MODULE: 24 | TYPE: 'GAP' 25 | 26 | CLASSIFIER: 27 | TYPE: 'LWS' 28 | BIAS: True 29 | 30 | LOSS: 31 | LOSS_TYPE: 'CrossEntropy' 32 | 33 | TRAIN: 34 | BATCH_SIZE: 128 35 | MAX_EPOCH: 30 36 | NUM_WORKERS: 16 37 | COMBINER: 38 | TYPE: 'default' 39 | OPTIMIZER: 40 | TYPE: 'SGD' 41 | BASE_LR: 0.1 42 | MOMENTUM: 0.9 43 | WEIGHT_DECAY: 5e-4 44 | LR_SCHEDULER: 45 | TYPE: 'warmup' 46 | LR_STEP: [10, 20] 47 | LR_FACTOR: 0.1 48 | WARM_EPOCH: 5 49 | TENSORBOARD: 50 | ENABLE: True 51 | SAMPLER: 52 | TYPE: 'weighted sampler' 53 | WEIGHTED_SAMPLER: 54 | TYPE: 'balance' 55 | 56 | DISTRIBUTED: False 57 | APEX: False 58 | SYNCBN: True 59 | 60 | TRANSFORMS: 61 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 62 | TEST_TRANSFORMS: () 63 | 64 | TEST: 65 | BATCH_SIZE: 64 66 | NUM_WORKERS: 8 67 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar10_im50/student.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive' 2 | OUTPUT_DIR: './output/cifar10_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'DiVEKLD' 27 | DiVEKLD: 28 | TEMPERATURE: 2.0 29 | POWER_NORM: True 30 | POWER: 0.5 31 | ALPHA: 0.5 32 | BASELOSS: 'BalancedSoftmaxCE' 33 | 34 | 35 | 36 | TRAIN: 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'dive' 42 | DIVE: 43 | TEACHER_MODEL: './output/cifar10_im50/dive_teacher/models/epoch_200.pth' 44 | 45 | OPTIMIZER: 46 | TYPE: 'SGD' 47 | BASE_LR: 0.1 48 | MOMENTUM: 0.9 49 | WEIGHT_DECAY: 2e-4 50 | LR_SCHEDULER: 51 | TYPE: 'warmup' 52 | LR_STEP: [120, 160] 53 | LR_FACTOR: 0.01 54 | WARM_EPOCH: 5 55 | 56 | TENSORBOARD: 57 | ENABLE: True 58 | SAMPLER: 59 | TYPE: 'default' 60 | 61 | DISTRIBUTED: False 62 | SYNCBN: True 63 | 64 | TRANSFORMS: 65 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 66 | TEST_TRANSFORMS: () 67 | 68 | TEST: 69 | BATCH_SIZE: 64 70 | NUM_WORKERS: 8 71 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar100_im50/student.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive' 2 | OUTPUT_DIR: './output/cifar100_im50' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.02 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'DiVEKLD' 27 | DiVEKLD: 28 | TEMPERATURE: 2.0 29 | POWER_NORM: True 30 | POWER: 0.5 31 | ALPHA: 0.5 32 | BASELOSS: 'BalancedSoftmaxCE' 33 | 34 | 35 | 36 | TRAIN: 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'dive' 42 | DIVE: 43 | TEACHER_MODEL: './output/cifar100_im50/dive_teacher/models/epoch_200.pth' 44 | 45 | OPTIMIZER: 46 | TYPE: 'SGD' 47 | BASE_LR: 0.1 48 | MOMENTUM: 0.9 49 | WEIGHT_DECAY: 2e-4 50 | LR_SCHEDULER: 51 | TYPE: 'warmup' 52 | LR_STEP: [120, 160] 53 | LR_FACTOR: 0.01 54 | WARM_EPOCH: 5 55 | 56 | TENSORBOARD: 57 | ENABLE: True 58 | SAMPLER: 59 | TYPE: 'default' 60 | 61 | DISTRIBUTED: False 62 | SYNCBN: True 63 | 64 | TRANSFORMS: 65 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 66 | TEST_TRANSFORMS: () 67 | 68 | TEST: 69 | BATCH_SIZE: 64 70 | NUM_WORKERS: 8 71 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar10_im100/student.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive' 2 | OUTPUT_DIR: './output/cifar10_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR10' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'DiVEKLD' 27 | DiVEKLD: 28 | TEMPERATURE: 2.0 29 | POWER_NORM: True 30 | POWER: 0.5 31 | ALPHA: 0.5 32 | BASELOSS: 'BalancedSoftmaxCE' 33 | 34 | 35 | 36 | TRAIN: 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'dive' 42 | DIVE: 43 | TEACHER_MODEL: './output/cifar10_im100/dive_teacher/models/epoch_200.pth' 44 | 45 | OPTIMIZER: 46 | TYPE: 'SGD' 47 | BASE_LR: 0.1 48 | MOMENTUM: 0.9 49 | WEIGHT_DECAY: 2e-4 50 | LR_SCHEDULER: 51 | TYPE: 'warmup' 52 | LR_STEP: [120, 160] 53 | LR_FACTOR: 0.01 54 | WARM_EPOCH: 5 55 | 56 | TENSORBOARD: 57 | ENABLE: True 58 | SAMPLER: 59 | TYPE: 'default' 60 | 61 | DISTRIBUTED: False 62 | SYNCBN: True 63 | 64 | TRANSFORMS: 65 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 66 | TEST_TRANSFORMS: () 67 | 68 | TEST: 69 | BATCH_SIZE: 64 70 | NUM_WORKERS: 8 71 | MODEL_FILE: 'best_model.pth' -------------------------------------------------------------------------------- /configs/cao_cifar/DiVE/cifar100_im100/student.yaml: -------------------------------------------------------------------------------- 1 | NAME: 'dive' 2 | OUTPUT_DIR: './output/cifar100_im100' 3 | SHOW_STEP: 100 4 | SAVE_STEP: 5 5 | VALID_STEP: 1 6 | INPUT_SIZE: (32, 32) 7 | COLOR_SPACE: 'RGB' 8 | 9 | 10 | DATASET: 11 | DATASET: 'IMBALANCECIFAR100' 12 | IMBALANCECIFAR: 13 | RATIO: 0.01 14 | 15 | BACKBONE: 16 | TYPE: 'res32_cifar' 17 | 18 | MODULE: 19 | TYPE: 'GAP' 20 | 21 | CLASSIFIER: 22 | TYPE: 'FC' 23 | BIAS: True 24 | 25 | LOSS: 26 | LOSS_TYPE: 'DiVEKLD' 27 | DiVEKLD: 28 | TEMPERATURE: 2.0 29 | POWER_NORM: True 30 | POWER: 0.5 31 | ALPHA: 0.5 32 | BASELOSS: 'BalancedSoftmaxCE' 33 | 34 | 35 | 36 | TRAIN: 37 | BATCH_SIZE: 128 38 | MAX_EPOCH: 200 39 | NUM_WORKERS: 16 40 | COMBINER: 41 | TYPE: 'dive' 42 | DIVE: 43 | TEACHER_MODEL: './output/cifar100_im100/dive_teacher/models/epoch_200.pth' 44 | 45 | OPTIMIZER: 46 | TYPE: 'SGD' 47 | BASE_LR: 0.1 48 | MOMENTUM: 0.9 49 | WEIGHT_DECAY: 2e-4 50 | LR_SCHEDULER: 51 | TYPE: 'warmup' 52 | LR_STEP: [120, 160] 53 | LR_FACTOR: 0.01 54 | WARM_EPOCH: 5 55 | 56 | TENSORBOARD: 57 | ENABLE: True 58 | SAMPLER: 59 | TYPE: 'default' 60 | 61 | DISTRIBUTED: False 62 | SYNCBN: True 63 | 64 | TRANSFORMS: 65 | TRAIN_TRANSFORMS: ('random_crop', 'random_horizontal_flip') 66 | TEST_TRANSFORMS: () 67 | 68 | TEST: 69 | BATCH_SIZE: 64 70 | NUM_WORKERS: 8 71 | MODEL_FILE: 'best_model.pth' --------------------------------------------------------------------------------