CondenseNet-PyTorch/configs/condensenet_exp_0.json at master · hagerrady13/CondenseNet-PyTorch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
{
  "exp_name": "exp_0",
  "agent": "CondenseNetAgent",
  "mode": "train",

  "cuda": true,
  "gpu_device": 0,
  "seed": 1337,

  "img_size": 32,
  "input_channels": 3,

  "num_classes": 10,

  "learning_rate": 0.1,
  "lr_type": "cosine",
  "weight_decay": 0.0001,
  "momentum": 0.9,

  "stages": [14,14,14],
  "growth_rate": [8,16,32],
  "conv_bottleneck": 4,
  "group1x1": 4,
  "group3x3": 4,
  "condense_factor": 4,
  "init_stride": 1,
  "pool_size": 8,
  "dropout_rate": 0,

  "batch_size": 64,
  "max_epoch": 300,

  "validate_every": 2,

  "checkpoint_file": "checkpoint.pth.tar",

  "data_loader": "Cifar10DataLoader",
  "data_loader_workers": 4,
  "pin_memory": true,
  "async_loading": true,

  "data_mode": "numpy_train",
  "data_folder": "./data/cifar10/numpy/",
  "x_train": "x_train.npy",
  "y_train": "y_train.npy",
  "x_valid": "x_trainval.npy",
  "y_valid": "y_trainval.npy",
  "x_test": "x_val.npy",
  "y_test": "y_val.npy"

}