deep-fashion-retrieval icon indicating copy to clipboard operation
deep-fashion-retrieval copied to clipboard

How to train? Please help

Open tussank opened this issue 6 years ago • 5 comments

Can you please explain how to train this on complete dataset? Your model is trained only on 20 categories. As per the instructions in the readme, I have to set freeze to true and train. Then, after 10 epochs. If I change freeze to False, then how to again train? When I try to train the same model, the accuracy moves from 59% to 3%? Can you please tell me, how to load my model for retrainig? Thanks

tussank avatar Jun 04 '18 08:06 tussank

The complete dataset? I just used two subset of them.

ihciah avatar Jun 04 '18 09:06 ihciah

I mean, the complete Attribute Prediction dataset which contains around 50 categories. Your trained model seems to be trained on only 20 categories.

tussank avatar Jun 04 '18 09:06 tussank

Yes, for convinence I only used upper clothes such as coat.

ihciah avatar Jun 04 '18 09:06 ihciah

I am trying to train it for 50 categories. I set the freeze to True and trained for 10 epochs. Then, as per your instruction, I have modified the config file. Now, how to retrain it? Is setting DUMPED_MODEL in config as that model and again starting to train enough? I am not able to understand what happens when we unfreeze the layers in the later step? will it affect all the layers of the model now?

tussank avatar Jun 04 '18 09:06 tussank

Yes, for convinence I only used upper clothes such as coat.

I wonder know which 20 categories did you selected (It seems that 20 upper clothes are not consist of coat.) Thanks!

whytin avatar Aug 19 '19 03:08 whytin