RE-OWOD/configs/OWOD/t2/t2_train_baseline_only_frcn...

16 lines
426 B
YAML

_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
MODEL:
WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_baseline/model_final.pth"
DATASETS:
TRAIN: ('t2_voc_coco_2007_train', )
TEST: ('voc_coco_2007_test', )
SOLVER:
STEPS: (50000, 60000)
MAX_ITER: 70000
WARMUP_ITERS: 0
OUTPUT_DIR: "./output/t2_baseline"
OWOD:
PREV_INTRODUCED_CLS: 20
CUR_INTRODUCED_CLS: 20
ENABLE_THRESHOLD_AUTOLABEL_UNK: False
ENABLE_CLUSTERING: False