File size: 279 Bytes
21c7db9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | """Training callbacks."""
from __future__ import annotations
from typing import Callable
def every_n_steps(n: int, fn: Callable[[int], None]) -> Callable[[int], None]:
def _callback(step: int) -> None:
if step % n == 0:
fn(step)
return _callback
|