utils for pytorch lightning
Project description
PL-utils
Utils for pytorch-lightning.
Collection of utils - datamodules, callbacks, models etc for pytorch-lighting.
Datamodules.
Imagenette.
Imagenette dataset Datamodule.
Subset of ImageNet.
https://github.com/fastai/imagenette
from pl_utils.imagenette_datamodule import ImagenetteDataModule, ImageWoofDataModule
imagenette_datamodule = ImagenetteDataModule(data_dir=DATADIR)
imagenette_datamodule.setup()
len(imagenette_datamodule.train_dataset), len(imagenette_datamodule.val_dataset)
(9469, 3925)
woof_datamodule = ImageWoofDataModule(data_dir=DATADIR)
woof_datamodule.setup()
len(woof_datamodule.train_dataset), len(woof_datamodule.val_dataset)
(9025, 3929)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pl_utils-0.0.1.tar.gz
(10.6 kB
view hashes)
Built Distribution
pl_utils-0.0.1-py3-none-any.whl
(10.1 kB
view hashes)