facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

Training recopies for the Weakly Supervised Pre-training paper #790

Closed netw0rkf10w closed 2 years ago

netw0rkf10w commented 2 years ago

📚 Documentation

Hello,

In the README you mentioned the following:

It offers several notable advantages:

...

I would like to know if you could share the training recipes (*.json) for the results that you were able to replicated.

Thank you in advance.

Best regards.

mannatsingh commented 2 years ago

Hi @netw0rkf10w, unfortunately this was trained a couple of years ago so it's hard to track the configs down for this. I'm afraid we won't be able to share this. But the training logic was fairly similar to training on imagenet, except that there were multiclass labels normalized to sum to 1. Other than that, everything should match up to imagenet 1k training with a cross entropy loss that handles soft targets.

mannatsingh commented 2 years ago

Please feel free to reopen if you have more questions