SingleViewReconstruction/CompressionAutoEncoder/config.json at main · DLR-RM/SingleViewReconstruction · 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
{
  "config": {
    "data": {
      "val_hard_samples_path": "",
      "truncation_threshold": 0.1,
      "data_augmentation": true,
      "train_hard_samples_path": "",
      "focused_blocks": 1,
      "eval_batch_size": 64,
      "batch_size": 256,
      "additional_border": 7,
      "additional_blocks": 0,
      "train_path": "/home/max/workspace/SingleViewReconstruction/data/*/voxelgrid",
      "val_path": "/home/max/workspace/SingleViewReconstruction/data/107*",
      "model_save_path": "/home/max/workspace/SingleViewReconstruction/data/ae_model",
      "block_size": 16,
      "data_size": 512
    },
    "model": {
      "decoder": {
        "output": {
          "padding": "same",
          "kernel_size": 3
        },
        "conv": {
          "default": {
            "learned_pool_upsampling": true,
            "upsampling_enabled": true,
            "activation": "relu",
            "padding": "same",
            "pooling_enabled": false,
            "batch_normalization": false,
            "kernel_size": 3
          },
          "1": {
            "channels": 64
          },
          "0": {
            "channels": 128
          },
          "3": {
            "channels": 16
          },
          "2": {
            "channels": 32
          },
          "4": {
            "channels": 64,
            "upsampling_enabled": false
          },
          "total": 5
        }
      },
      "encoder": {
        "conv": {
          "default": {
            "learned_pool_upsampling": true,
            "activation": "relu",
            "padding": "same",
            "pooling_enabled": true,
            "batch_normalization": false,
            "kernel_size": 3
          },
          "1": {
            "channels": 32,
            "padding": "valid"
          },
          "0": {
            "channels": 16,
            "padding": "valid"
          },
          "3": {
            "channels": 128
          },
          "2": {
            "channels": 64,
            "padding": "valid"
          },
          "4": {
            "channels": 64,
            "activation": "linear",
            "pooling_enabled": false
          },
          "total": 5
        }
      }
    },
    "optimizer": {
      "loss": "weighted_L1",
      "lr": 0.0003,
      "batch_topk_loss": 0,
      "topk_loss": 0
    },
    "checkpoint_interval": 0,
    "save_interval": 100
  }
}