ldctbench.methods.cnn10.Trainer

Trainer(args, device)

Bases: BaseTrainer

Trainer class for CNN10¹

---

1. H. Chen et al., “Low-dose CT via convolutional neural network,” Biomed Opt Express, vol. 8, no. 2, pp. 679–694, Jan. 2017. ↩

Parameters:

• args (Namespace) –

  Arguments to configure the trainer.

• device (device) –

  Torch device to use for training.