GantMan / GantMan/nsfw_model

fine tune do not change model output

Open
#33 7 comments 0 reactions 0 assignees View on GitHub
question training
Dominant language
Python
Stars
2.1k
Forks
303
PR merge metrics
No merged PRs in 30d

Description

Hi Thank you very much for your pre-trained model!

I am gathering some new data and using your code to fine-tune.After 100 epochs, I get a new model (one weird thing is that the size of new model reduce a lot.....). However when I test it, it seems that new model behave exactly same as the pre-trained one(same acc,same recall and same precision for each class).

Any idea why it happens? I am checking your code and seems that you only set one layer trainable?

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reviewing the repository's fine-tuning code and the configuration that determines which layer is trainable. Reproduce the reported result with the pre-trained and fine-tuned models, comparing model size and per-class accuracy, recall, and precision. Done means identifying why fine-tuning leaves the outputs unchanged and recording a confirmed fix or explanation.

Written by the indexing model from the issue text.

Assessment

Tech stack
keras, python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.