config_example.json

{
  "_MODEL": ["SGFN", "SGPN"],
  "MODEL": "SGFN",
  "VERBOSE": false,
  "SEED": 2020,
  "MAX_EPOCHES": 1,
  "LR": 0.001,
  "W_DECAY": false,
  "AMSGRAD":false,
  "_LR_SCHEDULE": ["None", "BatchSize"],
  "LR_SCHEDULE": "None",
  "GPU": [0],
  "SAVE_INTERVAL": 100,
  "VALID_INTERVAL": 5,
  "LOG_INTERVAL": 10,
  "LOG_IMG_INTERVAL": 100,
  "WORKERS": 4,
  "_EDGE_BUILD_TYPE": ["FC", "KNN"],
  "EDGE_BUILD_TYPE": "KNN",
  "WEIGHTING": true,
  "MODEL": {
    "N_LAYERS": 2,
    "USE_SPATIAL": true,
    "WITH_BN": false,
    "USE_GCN": true,
    "USE_RGB": false,
    "USE_NORMAL": false,
    "USE_CONTEXT": true,
    "USE_GCN_EDGE": true,
    "USE_REL_LOSS": true,
    "OBJ_PRED_FROM_GCN": true,
    "_GCN_TYPE": ["TRIP", "EAN"],
    "GCN_TYPE": "EAN",
    "_ATTENTION" : ["fat"],
    "ATTENTION": "fat",
    "DROP_OUT_ATTEN": 0.5,
    "multi_rel_outputs": true,
    "feature_transform": false,
    "point_feature_size": 256,
    "edge_feature_size":256,
    "gcn_hidden_feature_size": 512,
    "lambda_o": 0.1,
    "NUM_HEADS": 4,
    "_GCN_AGGR": ["add","mean","max"],
    "GCN_AGGR": "max",
    "DIM_ATTEN": 256,
    "_WEIGHT_EDGE": ["BG", "DYNAMIC","OCCU"],
    "WEIGHT_EDGE": "DYNAMIC",
    "w_bg": 1.0,
    "NONE_RATIO": 1.0
  },
  "dataset": {
    "root": ["./data/example_data"],
    "selection": "",
    "ignore_scannet_rel": true,
    "is_v2": true,
    "_label_file": ["labels.instances.align.annotated.v2.ply", "inseg.ply", "cvvseg.ply"],
    "label_file": "inseg.ply",
    "data_augmentation": false,
    "num_points": 128,
    "num_points_union": 256,
    "disable_support_rel": false,
    "with_bbox": false,
    "discard_some": false,
    "_dataset_type": ["SGPN", "SGFN"],
    "dataset_type": "SGFN",
    "load_cache": false,
    "sample_in_runtime": true,
    "sample_num_nn": 2,
    "sample_num_seed": 4,
    "class_choice": [],
    "max_edges": -1,
    "drop_edge": 0.5,
    "drop_edge_eval": 0.0
  }
}