Pre-training ANN model

[lanyx7]

Hello, I have tested the ANN model of VGG16 without usebn provided by you on ImageNet dataset, but the accuracy is only 55.074%. Could you please update the pre-training ANN model?

[yhhhli]

Do you use max pooling or average pooling in VGG?

[lanyx7]

I used avgpooling in VGG. I used the VGG model you provided from SNN_Calibration/ImageNet/models/vgg.py directly.

Do you use max pooling or average pooling in VGG?

[yhhhli]

Got it, I will double check the code, and update it in this weekend.

[lanyx7]

Got it, I will double check the code, and update it in this weekend.

Thank you for your reply. I also have tested the ANN model of VGG16 with BN and avgpooling provided by you on ImageNet dataset, it achieves an expected accuracy. Furthermore, I also tested the two ResNet34 pre-trained models you provided, which also achieved the expected accuracy. I hope these test results will help you.

[yhhhli]

just uploaded a new framework and new checkpoints, feel free to report any issues.