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/ensemble.py
+++ b/models/ensemble.py
@ -5,7 +5,7 @@ import torch
 from torch import nn
 from torch.nn import ModuleList

-from ml_lib.modules.utils import LightningBaseModule
+from ml_lib.modules.util import LightningBaseModule
 from ml_lib.utils.config import Config
 from ml_lib.utils.model_io import SavedLightningModels
 from util.module_mixins import (BaseOptimizerMixin, BaseTrainMixin, BaseValMixin, BinaryMaskDatasetMixin,