DataLoader forcing batches into __getitem__ of Dataset
I am trying to train a model for image segmentation using Pytorch Lightning. I created a custom Dataset used by a LightningDataModule. The custom Dataset loads a dataset from HuggingFace that contains images with associated masks (targets) and I mostly use this Dataset to apply the transformations I want for the model.
Load layers from compressed weights in pytorch
I am trying to do the following: