search

QDucasse / nn_benchmark

🧠 Benchmark facility to train networks on different datasets for PyTorch/Brevitas
MIT License
24 stars 1 forks source link

something wrong in MobilenetV1 #21

Closed imColin-Li closed 3 years ago

imColin-Li commented 3 years ago

I dont know the function of this code:

from brevitas.quant_tensor import pack_quant_tensor and The file is no longer in the source project. When i delete this code to run , I find a bug in forword process.

RuntimeError: The size of tensor a (64) must match the size of tensor b (32) at non-singleton dimension 1. when the program run to pw

QDucasse commented 3 years ago

Hi, The mobilenet version was a custom implementation at a time where it was not supported in FINN. I believe it has been implemented now and maybe you can take a look at https://github.com/Xilinx/finn-examples where an example is provided!

imColin-Li commented 3 years ago

Hi, Thank you very much for your reply! After I changed the network model, it is now working. Thank you for your advice.