steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SubSpectral and Lightning 0.9 Update

Browse Source
This commit is contained in:
Si11ium
2020-09-25 15:35:15 +02:00
parent 6bc9447ce1
commit 5848b528f0
13 changed files with 197 additions and 630 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
_templates/new_project/main.py
View File

@@ -7,10 +7,9 @@ import torch
from pytorch_lightning import Trainer
from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
from modules.utils import LightningBaseModule
from utils.config import Config
from utils.logging import Logger
from utils.model_io import SavedLightningModels
from ml_lib.modules.util import LightningBaseModule
from ml_lib.utils.config import Config
from ml_lib.utils.logging import Logger
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)

4
_templates/new_project/multi_run.py
View File

@@ -1,6 +1,6 @@
import warnings
from _templates.new_project.utils.project_config import Config
from ml_lib._templates.new_project.utils.project_config import Config
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@@ -8,7 +8,7 @@ warnings.filterwarnings('ignore', category=UserWarning)
# Imports
# =============================================================================
from _templates.new_project.main import run_lightning_loop, args
from ml_lib._templates.new_project.main import run_lightning_loop, args
if __name__ == '__main__':

10
_templates/new_project/utils/module_mixins.py
View File

@@ -11,13 +11,13 @@ from torch.utils.data import DataLoader
from torchcontrib.optim import SWA
from torchvision.transforms import Compose
from _templates.new_project.datasets.template_dataset import TemplateDataset
from ml_lib._templates.new_project.datasets.template_dataset import TemplateDataset
from audio_toolset.audio_io import NormalizeLocal
from modules.utils import LightningBaseModule
from utils.transforms import ToTensor
from ml_lib.audio_toolset.audio_io import NormalizeLocal
from ml_lib.modules.util import LightningBaseModule
from ml_lib.utils.transforms import ToTensor
from _templates.new_project.utils.project_config import GlobalVar as GlobalVars
from ml_lib._templates.new_project.utils.project_config import GlobalVar as GlobalVars
class BaseOptimizerMixin:

2
_templates/new_project/utils/project_config.py
View File

@@ -1,6 +1,6 @@
from argparse import Namespace
from utils.config import Config
from ml_lib.utils.config import Config
class GlobalVar(Namespace):