BATCH_SIZE = 16 PROPOSAL_NUM = 6 CAT_NUM = 4 INPUT_SIZE = (448, 448) # (w, h) LR = 0.001 WD = 1e-4 SAVE_FREQ = 1 resume = '' test_model = 'model.ckpt' save_dir = '/data_4t/yangz/models/'