nntoolbox.callbacks.resizing module

class nntoolbox.callbacks.resizing.InputProgressiveResizing(initial_size: int, max_size: int, upscale_every: int, upscale_factor: float, mode='bilinear')

Bases: nntoolbox.callbacks.callbacks.Callback

Implement a callback for progressive resizing (input only)

on_batch_begin(data: Dict[str, torch.Tensor], train) → Dict[str, torch.Tensor]

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