[VTA] ResNet-18 VTA offloading

Hello, everyone.

I have a question regarding deploy_vision_on_vta.py.

The code states that it offloads ResNet-18 part from [nn.max_pool2d] to [nn.global_avg_pool2d].

I want to how the 1st convolution before nn.max_pool2d is processed in VTA.

Thanks,

Jake

It isn’t.

AFAIK the 1st convolution will be performed on the Arm core which is in the FPGA fabric part of the Pynq target but not part of the FPGA fabric which is where the VTA is actually implemented.

1 Like

Thanks for responding.

Yes, you are right.

What I meant was how 1st convolution was done on the pynq board.

I wrote “VTA” instead of “pynq”, my bad.

Can you tell me the source code folder that helps to understand offloading part?

Thanks,

Jake

No problem.

I am not really up-to-date with the latest VTA codebase but taking an educated guess (and leaving some of the details for you to find) you can start here