CARE Module

Instantiate CARE Module.

Parameters:

• algorithm_config (CAREAlgorithm, N2NAlgorithm, or dict) –

  Configuration for the CARE algorithm, either as a CAREAlgorithm/N2NAlgorithm instance or a dictionary.

Configure optimizer and learning rate scheduler.

Returns:

• dict[str, Any] –

  A dictionary containing the optimizer and learning rate scheduler.

Forward pass.

Parameters:

• x (Tensor) –

  Input tensor.

Returns:

• Tensor –

  Model output tensor.

On fit start hook for CARE module.

Check that training and validation target data have been supplied.

Prediction step for CARE module.

Parameters:

• batch (ImageRegionData or (ImageRegionData, ImageRegionData)) –

  A tuple containing the input data and optionally the target data.

• batch_idx (int) –

  The index of the current batch in the prediction loop.

Returns:

• ImageRegionData –

  The output batch containing the predictions.

Training step for CARE module.

Parameters:

• batch ((ImageRegionData, ImageRegionData)) –

  A tuple containing the input data and the target data.

• batch_idx (int) –

  The index of the current batch in the training loop.

Returns:

• Tensor –

  The loss value computed for the current batch.

Validation step for CARE module.

Parameters:

• batch ((ImageRegionData, ImageRegionData)) –

  A tuple containing the input data and the target data.

• batch_idx (int) –

  The index of the current batch in the validation loop.