PyTorchのDatasetとDataloaderからBatchサイズ・Target・etc…を知りたい

PyTorchのDatasetとDataloader

DatasetとDataloaderの基本的な使い方と取得できる情報(属性)についてまとめます。

この記事は、取得できる情報(属性)についてが中心だと思ってください。
例えば、”dataloaderのBatchサイズが知りたい”とか、”dataloaderからdatasetを確認したい”ときに使います。

Dataset

https://betashort-lab.com/データサイエンス/ディープラーニング/pytorchのdatasetsで画像データセットを作る/

データやtargetを見る

len(dataloader)

dataset.data

dataset.targets

dataset.transform

Dataloader

https://betashort-lab.com/データサイエンス/ディープラーニング/pytorchのdataloader-samplerとclass_weightなども/

batchやdatasetの情報を見る

dataloader.dataset

dataloader.batch_size

len(dataloader)

dataloader.dataset.data

dataloader.dataset.targets

dataloader.dataset.transform