search

sachinraja13 / TabStructNet

Other
127 stars 31 forks source link

Fine tune model #7

Closed gauravlochab closed 3 years ago

gauravlochab commented 3 years ago

Could you tell the best way to fine tune the model on my custom dataset at different training levels ? 'Finetune layers from ResNet stage 4 and up' 'Fine tune all layers' these comments in tabnet.py are unclear Thanks for the amazing work appreciate your prompt reply!

sachinraja13 commented 3 years ago

Hi,

I would recommend training heads for 20 epochs, training ResNet stage 4 and up for another 20 epochs, and training all layers for additional 40 epochs. You can have a look at how loss is decreasing and can appropriately change learning rate in the config.py file.

Please make sure to execute "python setup.py install" after you have made any changes to the model.py or config.py file in the mrcnn folder.

Hope this helps.

gauravlochab commented 3 years ago

thanks, it helped

prati2a22 commented 2 years ago

thanks, it helped

Hey @gauravlochab, can you share how you annotated the data for finetuning? Which specific tool you used for annotation? I keep getting a KeyError for 'start_row' as my annotator tool did not generate the start_row, start_col and end_row, end_col fields.

gauravlochab commented 2 years ago

Hi @prati2a22 , i cant share the data with but you can easily annotate the data as shown in the tabstructnet paper. i used cvat for annotating the images.

prati2a22 commented 2 years ago

Hi @prati2a22 , i cant share the data with but you can easily annotate the data as shown in the tabstructnet paper. i used cvat for annotating the images.

Hey thanks for the info @gauravlochab, I checked out the CVAT documentation and I was wondering how did you add the "start_row", "start_col", "end_row", "end_col" fields. That's the part I'm struggling with.