forked from andriluka/ReInspect
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
43 lines (43 loc) · 1.17 KB
/
config.json
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
{
"data": {
"test_idl": "./data/brainwash/brainwash_val.idl",
"train_idl": "./data/brainwash/brainwash_train.idl",
"idl_mean": "./data/brainwash_mean.npy"
},
"solver": {
"base_lr": 0.2,
"max_iter": 800001,
"test_iter": 100,
"test_interval": 1000,
"momentum": 0.5,
"random_seed": 2,
"clip_gradients": 0.1,
"gamma": 0.8,
"stepsize": 100000,
"weights": ""
},
"logging": {
"display_interval": 100,
"snapshot_prefix": "/tmp/reinspect",
"snapshot_interval": 10000,
"schematic_path": "/tmp/reinspect.png",
"graph_prefix": "/tmp/history",
"log_file": "/tmp/log_brainwash.txt",
"graph_interval": 500
},
"net": {
"init_range": 0.1,
"max_len": 5,
"lstm_num_cells": 250,
"dropout_ratio": 0.15,
"hungarian_match_ratio": 0.5,
"hungarian_permute_matches": true,
"hungarian_loss_weight": 0.03,
"googlenet_lr_mult": 4,
"grid_height": 15,
"grid_width": 20,
"img_height": 480,
"img_width": 640,
"region_size": 64
}
}