ResNet-50 Python Brevitas Model

[oshears]

Hi FINN Team,

I am currently working on a project that is interested in recreating the quantized .onnx file used for the ResNet-50 FINN example here:

https://github.com/Xilinx/finn-examples/tree/main/build/resnet50

Is there any way that the notebook/python code for creating the quantized, brevitas version of the ResNet-50 model can be shared?

I am interested in trying to adapt the network to fit on a ZCU102/ZCU104.

[mmrahorovic]

Hi @oshears, good question and great to hear about your project!
The Brevitas code for ResNet-50 is indeed currently not released, but we are working on it and we will let you know in the upcoming months when it is ready!

[MRFiruzabadi]

Hi @mmrahorovic, Did you manage to release the Brevitas code for ResNet-50?

[mmrahorovic]

Hi @MRFiruzabadi
The code isn't yet released. However, in the meantime, you can find the ResNet-18 model in Brevitas, which can serve as a starting point for extending this to a ResNet-50 model: https://github.com/Xilinx/brevitas/blob/dev/src/brevitas_examples/bnn_pynq/models/resnet.py.

[Adib6637]

Hi, has the code for resnet 50 been released?