pytorch stubs Dataset ¶ Bases: Generic[T_co] Dataset class stub. DataLoader ¶ Bases: Generic[T_co] DataLoader class stub. LightningModule ¶ LightningModule class stub. nn ¶ Pytorch nn class stub. Module ¶ Pytorch nn.Module class stub. torch ¶ Pytorch class stub. Was this page helpful? Thanks for your feedback! Thanks for your feedback! Help us improve this page by using our feedback form.