Available Modules
train
predict
utils.config
utils.custom_classes.GarbageClassifier
utils.custom_classes.GarbageDataModule
utils.custom_classes.LossCurveCallback