steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Audio Dataset

Browse Source
This commit is contained in:
Si11ium
2020-12-01 16:37:16 +01:00
parent 95561acc35
commit 95dcf22f3d
15 changed files with 468 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/ensemble.py
View File

@ -8,11 +8,11 @@ from torch.nn import ModuleList
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,
from util.module_mixins import (BaseOptimizerMixin, BaseTrainMixin, BaseValMixin, DatasetMixin,
BaseDataloadersMixin)
class Ensemble(BinaryMaskDatasetMixin,
class Ensemble(DatasetMixin,
BaseDataloadersMixin,
BaseTrainMixin,
BaseValMixin,