microsoft / unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

Home Page:https://aka.ms/GeneralAI

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Finetune DiT on custom dataset

jigsawcoder opened this issue · comments

I am using DiT-Base Cascade R-CNN trained on ICDAR 2019 cTDaR (Table Detection) model to finetune my custom dataset which has 2 classes only.

How do I modify config files to finetune this model on 2 classes??

Thanks in advance.