v1.0.1

UPDATE:
New basel model has a new architecture, using Residual Blocks inspired by ResNet.

Please use this new model. The old one can still be used, but in order to correctly work, you will have to firstly rename it to main_model.pt, then search within the project all the import lines of MainCNN:

from src.models.main_model.resnet_inspired_model_block import MainCNN

and update them to import from the old model class:

from src.models.main_model.model_class imaport MainCNN