search

gpleiss / temperature_scaling

A simple way to calibrate your neural network.
MIT License
1.09k stars 159 forks source link

Memory issue #9

Open skynbe opened 6 years ago

skynbe commented 6 years ago

I got a problem when I use more than 50000 images for validation set to calculate the optimal temperature. Is there any way to bypass this problem?

gpleiss commented 5 years ago

What step is running out of memory? This should be a solvable problem. Also, you probably don't need that many validation set images for temperature scaling. A few hundred should suffice (you are only learning a single parameter).