Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

structure fine-tuning issue #181

Open
tzktok opened this issue May 16, 2024 · 2 comments
Open

structure fine-tuning issue #181

tzktok opened this issue May 16, 2024 · 2 comments

Comments

@tzktok
Copy link

tzktok commented May 16, 2024

Hi, I am trying to fine tune the the table structure over on TATR-v1.1-All-msft.pth model...In my dataset i provide an 50 images for finetune...i am getting below training metrics...

(tables-detr) PS E:\table-transformer\src> python main.py --data_type structure --config_file structure_config.json --data_root_dir data_set --model_load_path TATR-v1.1-All-msft.pth --epochs 50  --mode train --model_save_dir fina
l_model --load_weights_only

pubmed: AP50: 0.811, AP75: 0.637, AP: 0.602, AR: 0.713
Total training time:  0:02:20.607232

How to increase the training accuracy more...i am using default structure config..In dataset how many images needed to get more better result..

@bsmock @dnfclas

@YingxuanW
Copy link

I just saw someone use 1k datas to fine-tuning the detection model.

@YingxuanW
Copy link

Hi, I am trying to fine tune the the table structure over on TATR-v1.1-All-msft.pth model...In my dataset i provide an 50 images for finetune...i am getting below training metrics...

(tables-detr) PS E:\table-transformer\src> python main.py --data_type structure --config_file structure_config.json --data_root_dir data_set --model_load_path TATR-v1.1-All-msft.pth --epochs 50  --mode train --model_save_dir fina
l_model --load_weights_only
pubmed: AP50: 0.811, AP75: 0.637, AP: 0.602, AR: 0.713
Total training time:  0:02:20.607232

How to increase the training accuracy more...i am using default structure config..In dataset how many images needed to get more better result..

@bsmock @dnfclas

Can you please tell me how did you prepare your fine-tuning datas? I also need to fine-tune thee table-structure model!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants