3 ms·
You can always decrease a batch size to fit in smaller RAM.
by solomatov 8y ago
You can always decrease a batch size to fit in smaller RAM.
- bitL 8y agoUnless you are already on a batch size 1 and TensorFlow greets you with a nice OOM message... Try some Wide-ResNet for state-of-art classification and play with its widening parameter.
- p1esk 8y agoUse gradient checkpointing.
- rspeer 8y ago...if the batch is the thing that doesn't fit in RAM, as opposed to the model.