Skip to content

ldctbench.methods.bilateral.Trainer

Trainer(args, device)

Bases: BaseTrainer

Trainer class for trainable bilateral filter1

  1. F. Wagner et al., “Ultralow-parameter denoising: Trainable bilateral filter layers in computed tomography,” Medical Physics, vol. 49, no. 8, pp. 5107–5120, 2022. ↩

Parameters:

  • args (Namespace) –

    Arguments to configure the trainer.

  • device (device) –

    Torch device to use for training.