This repository contains an official implementation of the Neural Bootstrapper.
- python==3.7
- pytorch==1.2
- torchvision
- tqdm
- PIL
Before running main.py, please make an ini script file under the script folder for configurations.
Example of ini_file:
[default]
dataset = cifar100
output_dir = outs
num_epoch = 200
dist = False
phase = train
cpus = 4
gpus = '2'
model = resnet34
is_nbs = True
num_classes = 100
lr = 0.1
weight_decay = 0.0005
optim = sgd
batch_size = 128
n_a = 400
num_bs = 100
dropout_rate = 0.
scheduler = cosine
epoch_th = 30➜ python main.py cutout/cifar10➜ python main.py cutout/cifar100➜ python main.py cutout/svhn