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

Relevance analysis has been moved out of predict.py and into a separate
script. It is not currently functional.