sith-experiments/param-files/Deep-LP/MNIST/rotation_test/standard.yaml at main · skarukas/sith-experiments · 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
52
53
54
55
model:
  type: logpolar
  layer_params:
    # layer 1
    - in_channels: 1
      out_channels: 32
      tau_min: 1
      tau_max: 30
      ntau: 20
      num_angles: 12
      topk: 12
      kernel_size: 7
      lp_version: bilinear
    # layer 2
    - in_channels: 32
      out_channels: 32
      tau_min: 1
      tau_max: 30
      ntau: 20
      num_angles: 12
      topk: 12
      kernel_size: 5
      lp_version: bilinear
    # layer 3
    - in_channels: 32
      out_channels: 32
      tau_min: 1
      tau_max: 30
      ntau: 20
      num_angles: 12
      topk: 12
      kernel_size: 5
      lp_version: bilinear
  dropout: 0.2
  batch_norm: False
  act_func: relu
  out_classes: 10
  output: max
optimizer:
  type: adam
  params:
    lr: 0.001
num_epochs: 40
batch_size: 16
train_data_dir:
  type: FastMNIST
  kwargs:
    root: "data"
    train: True
val_data_dir:
  type: FastMNIST
  kwargs:
    root: "data"
    train: False
collate: batch