dataflowr / Project-age-estimation-pytorch

PyTorch-based CNN implementation for estimating age from face images
4 stars 10 forks source link

Label smoothing / Faster training / Gaussian/Laplace loss #1

Open cnd3617 opened 3 years ago

cnd3617 commented 3 years ago

Dear All,

We are investigating (and trying to implement) label smoothing for the classification and regression task. The goal is to add regularization elements to the model, consequently reducing the MAE.

We are also implementing a new way of training the model by pre-computing features. The goal is to obtain a trainable (i.e. faster) model for non-using-GPU computers.

Finally, we are going to work on implementing a Gaussian/Laplace loss.

Amaury, Malo, & Côme

abursuc commented 3 years ago

Thank you so much for the update. Don't hesitate to reach out in case there are blocking or unclear points. Good luck!