---
title: SpikeDetection
canonical: >-
  https://lightning.ai/docs/pytorch/stable/optional-api/api/lightning.pytorch.callbacks.SpikeDetection
---
# SpikeDetection {#spikedetection}

### `class lightning.pytorch.callbacks.SpikeDetection(mode='min', window=10, warmup=1, atol=None, rtol=2.0, exclude_batches_path=None, finite_only=True)` {#lightning.pytorch.callbacks.SpikeDetection}

Bases: `SpikeDetection`, [`Callback`](/docs/pytorch/2.6.6/optional-api/api/lightning.pytorch.callbacks.Callback.md#lightning.pytorch.callbacks.Callback "lightning.pytorch.callbacks.callback.Callback")

#### `on_train_batch_end(trainer, pl_module, outputs, batch, batch_idx)` {#lightning.pytorch.callbacks.SpikeDetection.on_train_batch_end}

Checks if we currently have a loss-spike.

Return type:

[`None`](https://docs.python.org/3/builtins/constants.html#None "(in Python v3.15)")

---

**Navigation**

- Parent: [Api](/docs/pytorch/2.6.6/optional-api/api.md)