<filename>src/linearized_nns/pytorch_impl/__init__.py
from linearized_nns.pytorch_impl.classifier_training import ClassifierTraining
<filename>src/linearized_nns/pytorch_impl/__init__.py
from linearized_nns.pytorch_impl.classifier_training import ClassifierTraining