config = {}
config['CUDA_VISIBLE_DEVICES'] = "0"
config['train_root'] = './dataset/train/'
config['augmentation_prob'] = 0.4
config['batch_size'] = 2
config['test_batch_size'] = 1
config['cuda'] = 'True'
config['loss_type'] = 'ce'
config['nclass'] = 1
config['filter_multiplier'] = 8
config['block_multiplier'] = 5
config['num_layers'] = 9
config['step'] = 5
config['lr'] = 0.025
config['momentum'] = 0.9
config['weight_decay'] = 3e-4
config['arch_lr'] = 3e-3
config['arch_weight_decay'] = 1e-3
config['lr_scheduler'] = 'cos'
config['epochs'] = 40
config['min_lr'] = 0.001
config['alpha_epoch'] = config['epochs'] // 4
config['num_class'] = 2
config['steps'] = 4