search

Xilinx / brevitas

Brevitas: neural network quantization in PyTorch
https://xilinx.github.io/brevitas/
Other
1.21k stars 197 forks source link

how do you train MobileNet? #387

Open laochonlam opened 2 years ago

laochonlam commented 2 years ago

Thanks for releasing the code and your effort. I like this project!

My goal is to compile the Brevitas MobileNet model with my own dataset trained then deploy it to FINN. However, I couldn't find out how you train/retrain the MobileNet model but this issue (https://github.com/Xilinx/brevitas/issues/42).

I am wondering if you still plan to release the MobileNet training script? Also, I am not quite sure If I understood your reply to that issue:

It seems that you used the Brevitas MobileNet to fine-tune the pre-trained floating-point MobileNet weights got from here? If so, I assume the learning rate or other training parameters should be quite different from normal training. It would be great if I can get any tutorial from your side.

Sincerely, ChonLam

Ba1tu3han commented 1 year ago

Hello,

Did you find how to train the model?