Feb. 2, 2024, 9:27 p.m. | /u/rakk109

Computer Vision www.reddit.com

I was using pytorch's implementation of VGG16 with the pretrained weights I get when weights='DEFAULT'. i.e is the imagenet1k weights.

These were used to train the model on the cifar100, just trained for 1 epoch to verify and then evaluated on the testset

What I found is weird when the batch size is 12 the model gives a test accuracy of 20% (which I found unsettling since I was using the pretrained weights)

I then test on batch size of …

computervision found impact implementation performance pytorch train verify

Data Engineer

@ Lemon.io | Remote: Europe, LATAM, Canada, UK, Asia, Oceania

Artificial Intelligence – Bioinformatic Expert

@ University of Texas Medical Branch | Galveston, TX

Lead Developer (AI)

@ Cere Network | San Francisco, US

Research Engineer

@ Allora Labs | Remote

Ecosystem Manager

@ Allora Labs | Remote

Founding AI Engineer, Agents

@ Occam AI | New York