Skip to content

ldctbench.methods.qae.Trainer

Trainer(args, device)

Bases: BaseTrainer

Trainer class for Quadratic Autoencoder (QAE)1.


  1. F. Fan et al., “Quadratic autoencoder (Q-AE) for low-dose CT denoising,” IEEE Transactions on Medical Imaging, vol. 39, no. 6, pp. 2035–2050, Jun. 2020. 

Parameters:

  • args (Namespace) –

    Arguments to configure the trainer.

  • device (device) –

    Torch device to use for training.