cabs_gauge

class cockpit.instruments.cabs_gauge(self, fig, gridspec)

CABS gauge, showing the CABS rule versus iteration.

The batch size trades-off more accurate gradient approximations with longer computation. The CABS criterion describes the optimal batch size under certain assumptions.

The instruments shows the suggested batch size (and an exponential weighted average) over the course of training, according to

• Balles, L., Romero, J., & Hennig, P., Coupling adaptive batch sizes with learning rates (2017).

Preview

Requires

This instrument requires data from the CABS quantity class.

Parameters

• self (CockpitPlotter) – The cockpit plotter requesting this instrument.

• fig (matplotlib.figure.Figure) – Figure of the Cockpit.

• gridspec (matplotlib.gridspec.GridSpec) – GridSpec where the instrument should be placed