configs/Base_Omni3D.yaml

_BASE_: "Base.yaml"
SOLVER:
  TYPE: "sgd"
  IMS_PER_BATCH: 2 #196 -> r=5,6 -> because of dataset size r=5,6 * 10,335/233 = 0,248
  BASE_LR: 0.0214 #0.12
  STEPS: (17280, 23040)
  MAX_ITER: 100000 #116000
  WARMUP_ITERS: 0 #3625
TEST:
  EVAL_PERIOD: 7200 #29000
VIS_PERIOD: 1 #2320
DATASETS:
  TRAIN: ('SUNRGBD_train_mini_equalised', 'SUNRGBD_val_mini_equalised')
  TEST: ('SUNRGBD_test',) 
  CATEGORY_NAMES: ('chair', 'table', 'cabinet', 'car', 'lamp', 'books', 'sofa', 'pedestrian', 'picture', 'window', 'pillow', 'truck', 'door', 'blinds', 'sink', 'shelves', 'television', 'shoes', 'cup', 'bottle', 'bookcase', 'laptop', 'desk', 'cereal box', 'floor mat', 'traffic cone', 'mirror', 'barrier', 'counter', 'camera', 'bicycle', 'toilet', 'bus', 'bed', 'refrigerator', 'trailer', 'box', 'oven', 'clothes', 'van', 'towel', 'motorcycle', 'night stand', 'stove', 'machine', 'stationery', 'bathtub', 'cyclist', 'curtain', 'bin')
MODEL:
  ROI_HEADS:
    NUM_CLASSES: 50
  DEVICE: 'cpu' # forced to cpu in inference for now