fingerprinted now should work correctly

2020-05-19 08:33:05 +02:00
parent f57e25efdc
commit c083207235
10 changed files with 65 additions and 14 deletions
--- a/models/conv_classifier.py
+++ b/models/conv_classifier.py
@@ -4,7 +4,7 @@ from torch import nn
 from torch.nn import ModuleList

 from ml_lib.modules.blocks import ConvModule, LinearModule
-from ml_lib.modules.utils import LightningBaseModule
+from ml_lib.modules.util import LightningBaseModule
 from util.module_mixins import (BaseOptimizerMixin, BaseTrainMixin, BaseValMixin, BinaryMaskDatasetMixin,
                                BaseDataloadersMixin)