Deep-Learning-with-PyTorch

Training a
classification model
to detect suspected tumors
This chapter covers
• Using PyTorch DataLoaders to load data
• Implementing a model that performs
classification on our CT data
• Setting up the basic skeleton for our application
• Logging and displaying metrics
In the previous chapters, we set the stage for our cancer-detection project. We covered
medical details of lung cancer, took a look at the main data sources we will use
for our project, and transformed our raw CT scans into a PyTorch Dataset
instance. Now that we have a dataset, we can easily consume our training data. So
let’s do that!
279