rivet/sample_options.json at master · bvraghav/rivet · 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
{
  "cuda"                : true,
  "num_epochs"          : 200,
  "report_frequency"    : 32,
  "save_frequency"      : 1,

  "network"             : "resnet18",
  "weights_file"        : "/home/bvr/data/pytocone/save-20180427-114144/model_best.pth.tar",
  "weights_key"         : "state_dict",

  "model"               : "feature_pair",
  "model_params"        : {
    "fc"                : null,
    "feat_len"          : 512
  },

  "dataset"             : "pairwise",
  "dataset_params"      : {
    "adjacency"         : "/path/to/adjacency.json",
    "image_list"        : "/path/to/image_list.json",
    "labels"            : [0, 1],
    "transform"         : "sketch_transform"
  },

  "dataloader_params"   : {
    "batch_size"        : 256,
    "shuffle"           : true,
    "num_workers"       : 7
  },

  "criterion"           : "contrastive",
  "criterion_params"    : {
    "distance"          : "euclidean" ,
    "margin"            : 2.0
  },

  "optimizer"           : "Adam",
  "optimizer_params"    : {
    "weight_decay"      : 0.1
  },

  "lr_adjuster"         : "ReduceLROnPlateau",
  "lr_adjuster_params"  : {
    "eps"                 : 1e-4
  },

  "accuracy_transform"        : "loss_interpreter",
  "accuracy_transform_params" : {},

  "save_location"       : ".",
  "saver_current"       : "checkpoint.pth.tar",
  "saver_best"          : "model_best.pth.tar",

  "reporters"           : [ "log_average"
                          ]

}