Skip to content

ldctbench.methods.redcnn.Trainer

Trainer(args, device)

Bases: BaseTrainer

Trainer class for RED-CNN1

  1. H. Chen et al., “Low-dose CT with a residual encoder-decoder convolutional neural network,” IEEE Transactions on Medical Imaging, vol. 36, no. 12, pp. 2524–2535, Dec. 2017. ↩

Parameters:

  • args (Namespace) –

    Arguments to configure the trainer.

  • device (device) –

    Torch device to use for training.