IOBenchTask# class torchgeo.trainers.IOBenchTask[source]# Bases: BaseTask I/O benchmarking. Added in version 0.6. configure_models()[source]# No-op. configure_optimizers()[source]# Initialize the optimizer. Returns: Optimizer. Return type: Optimizer | Sequence[Optimizer] | tuple[Sequence[Optimizer], Sequence[LRScheduler | ReduceLROnPlateau | LRSchedulerConfig]] | OptimizerConfig | OptimizerLRSchedulerConfig | Sequence[OptimizerConfig] | Sequence[OptimizerLRSchedulerConfig] | None training_step(batch, batch_idx, dataloader_idx=0)[source]# No-op. Parameters: batch (dict[str, Any]) – The output of your DataLoader. batch_idx (int) – Integer displaying index of this batch. dataloader_idx (int) – Index of the current dataloader. Returns: Zero. Return type: Tensor validation_step(batch, batch_idx, dataloader_idx=0)[source]# No-op. Parameters: batch (dict[str, Any]) – The output of your DataLoader. batch_idx (int) – Integer displaying index of this batch. dataloader_idx (int) – Index of the current dataloader. test_step(batch, batch_idx, dataloader_idx=0)[source]# No-op. Parameters: batch (dict[str, Any]) – The output of your DataLoader. batch_idx (int) – Integer displaying index of this batch. dataloader_idx (int) – Index of the current dataloader. predict_step(batch, batch_idx, dataloader_idx=0)[source]# No-op. Parameters: batch (dict[str, Any]) – The output of your DataLoader. batch_idx (int) – Integer displaying index of this batch. dataloader_idx (int) – Index of the current dataloader.
