nntoolbox.callbacks.bs_scheduler module

class nntoolbox.callbacks.bs_scheduler.BatchSizeScheduler(train_data: torch.utils.data.dataloader.DataLoader, bs_schedule_fn: Callable[[int], int], timescale: str = 'iter')[source]

Bases: nntoolbox.callbacks.callbacks.Callback

on_batch_end(logs: Dict[str, Any])[source]
on_epoch_end(logs: Dict[str, Any]) → bool[source]