search

Raschka-research-group / coral-cnn

Rank Consistent Ordinal Regression for Neural Networks with Application to Age Estimation
https://www.sciencedirect.com/science/article/pii/S016786552030413X
MIT License
341 stars 62 forks source link

Pretrained model #1

Closed marcomiglionico94 closed 5 years ago

marcomiglionico94 commented 5 years ago

Hi, do you have any pretrained model so that it is possible to do inference using the network, instead of training everything from scratch? Thanks

rasbt commented 5 years ago

in the beginning, i saved all the networks but due to the large file size, I later just saved the predictions and probabilities on the test set. I think i should still have them on my external backup drive. which one are you interested in? Btw training takes "only" 1-4 hours depending on the dataset.

rasbt commented 5 years ago

Just found them. You can get the pt files from here: https://drive.google.com/drive/folders/199T9UZkMgaRSikTW40J0KH-0pJwzilpK?usp=sharing

(they may only work for PyTorch 1.0 though).

marcomiglionico94 commented 5 years ago

Thanks!! @rasbt