search

Neurobiologist / EC601-Pulmonary-Embolism

Development of a Deep Learning Model to Diagnose Pulmonary Embolism
3 stars 2 forks source link

Verify Sort: ct_data_loader.py #16

Closed Neurobiologist closed 3 years ago

Neurobiologist commented 3 years ago

ct_data_loader.py:

""" Base class DataLoader for loading a 3d dataset. This data loader is designed to work with
sequential models, and takes care of sorting batches and padding them for the pytorch
recurrent networks. Note that the dataset MUST BE SORTED BY LENGTH for this to work."""

Originally posted by @Neurobiologist in https://github.com/Neurobiologist/EC601-Pulmonary-Embolism/issues/14#issuecomment-716634405

Neurobiologist commented 3 years ago

Sort confirmed.