Logan Mo

Try a smaller learning rate?

> Hi, do you happen to remember what accuracy you achieved with efficientnet? What did you get?

1) In main.py right after transforms and before actually creating a model, replace that code with your own dataset class and its data-loader, followed by classes in your dataset. 2)...

@bryanbocao Your device is not being set to the GPU. Can you make sure if your Cuda drivers are properly installed and all models and datasets are being sent to...

> @Phillibob55 That's a different problem. The current issue is not about CUDA driver installation/configurations. I can run it on GPU but I intentionally wanted to test the CPU runtime,...

@bryanbocao Can you kinda guide to make these models work with image sizes other than 32x32?

@bryanbocao I started off with the first approach and just added a resize transform, but that loses a lot of information. For datasets like ImageNet, this doesn't give accuracy above...

I've created a repo for it [here](https://github.com/Phillibob55/image-classification)