search

alexmr09 / Mixed-precision-Neural-Networks-on-RISC-V-Cores

13 stars 2 forks source link

How is BatchNorm implemented in CNNs? #3

Closed Jzjerry closed 1 month ago

Jzjerry commented 1 month ago

Hi,

I didn't find any code related to batch norm layers in the inference code, and I'd like to know how they are handled. Are they fused with Conv2D layers?

alexmr09 commented 1 month ago

Currently, the batch normalization layers are fused with the convolutional layers before proceeding with fine-tuning the model's parameters.

Jzjerry commented 1 month ago

Got it, thanks!