Implemented BCEWithLogitsLoss reweighting function
Removed save_hyperparameters from model
Apply augmentation transforms on singular images
Fixed model summary

[datamodule] Slightly streamlines the datamodule approach; adds documentation; adds type annotations; adds TODOs

Created base class for every LightningDataModule, that can be inherited
from. For each protocol, the user should create a new class inheriting
from BaseDataModule and implement the setup() function, in which the
dataset is defined.

DataModule is generic and not tied to a specific dataset.