Spaces:

nielsr
/

dit-document-layout-analysis

Runtime error

dit-document-layout-analysis / cascade_dit_base.yml

nielsr HF staff

Update cascade_dit_base.yml

00a1e29 about 2 years ago

No virus

486 Bytes

	_BASE_: "Base-RCNN-FPN.yml"
	MODEL:
	PIXEL_MEAN: [ 127.5, 127.5, 127.5 ]
	PIXEL_STD: [ 127.5, 127.5, 127.5 ]
	WEIGHTS: "https://layoutlm.blob.core.windows.net/dit/dit-pts/dit-base-224-p16-500k-62d53a.pth"
	VIT:
	NAME: "dit_base_patch16"
	ROI_HEADS:
	NAME: CascadeROIHeads
	ROI_BOX_HEAD:
	CLS_AGNOSTIC_BBOX_REG: True
	RPN:
	POST_NMS_TOPK_TRAIN: 2000
	SOLVER:
	WARMUP_ITERS: 1000
	IMS_PER_BATCH: 16
	MAX_ITER: 60000
	CHECKPOINT_PERIOD: 2000
	TEST:
	EVAL_PERIOD: 2000