Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename to mayamodel

This commit is contained in:
shahules786 2022-11-15 14:29:04 +05:30
parent ba63c54399
commit bfd53937c2
5 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mayavoz/models/__init__.py
View File

@ -1,3 +1,3 @@
from mayavoz.models.demucs import Demucs from mayavoz.models.demucs import Demucs
from mayavoz.models.model import Model from mayavoz.models.model import Mayamodel
from mayavoz.models.waveunet import WaveUnet from mayavoz.models.waveunet import WaveUnet

4
mayavoz/models/dccrn.py
View File

@ -6,7 +6,7 @@ import torch.nn.functional as F
from torch import nn from torch import nn
from mayavoz.data import EnhancerDataset from mayavoz.data import EnhancerDataset
from mayavoz.models import Model from mayavoz.models import Mayamodel
from mayavoz.models.complexnn import ( from mayavoz.models.complexnn import (
ComplexBatchNorm2D, ComplexBatchNorm2D,
ComplexConv2d, ComplexConv2d,
@ -98,7 +98,7 @@ class DCCRN_DECODER(nn.Module):
return self.decoder(waveform) return self.decoder(waveform)
class DCCRN(Model): class DCCRN(Mayamodel):
STFT_DEFAULTS = { STFT_DEFAULTS = {
"window_len": 400, "window_len": 400,

4
mayavoz/models/demucs.py
View File

@ -6,7 +6,7 @@ import torch.nn.functional as F
from torch import nn from torch import nn
from mayavoz.data.dataset import EnhancerDataset from mayavoz.data.dataset import EnhancerDataset
from mayavoz.models.model import Model from mayavoz.models.model import Mayamodel
from mayavoz.utils.io import Audio as audio from mayavoz.utils.io import Audio as audio
from mayavoz.utils.utils import merge_dict from mayavoz.utils.utils import merge_dict
@ -88,7 +88,7 @@ class DemucsDecoder(nn.Module):
return out return out
class Demucs(Model): class Demucs(Mayamodel):
""" """
Demucs model from https://arxiv.org/pdf/1911.13254.pdf Demucs model from https://arxiv.org/pdf/1911.13254.pdf
parameters: parameters:

6
mayavoz/models/model.py
View File

@ -27,7 +27,7 @@ DEFAULT_DEVICE = "cpu"
SAVE_NAME = "enhancer" SAVE_NAME = "enhancer"
class Model(pl.LightningModule): class Mayamodel(pl.LightningModule):
""" """
Base class for all models Base class for all models
parameters: parameters:
@ -288,8 +288,8 @@ class Model(pl.LightningModule):
Returns Returns
------- -------
model : Model model : Mayamodel
Model Mayamodel
See also See also
-------- --------

4
mayavoz/models/waveunet.py
View File

@ -6,7 +6,7 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from mayavoz.data.dataset import EnhancerDataset from mayavoz.data.dataset import EnhancerDataset
from mayavoz.models.model import Model from mayavoz.models.model import Mayamodel
class WavenetDecoder(nn.Module): class WavenetDecoder(nn.Module):
@ -66,7 +66,7 @@ class WavenetEncoder(nn.Module):
return self.encoder(waveform) return self.encoder(waveform)
class WaveUnet(Model): class WaveUnet(Mayamodel):
""" """
Wave-U-Net model from https://arxiv.org/pdf/1811.11307.pdf Wave-U-Net model from https://arxiv.org/pdf/1811.11307.pdf
parameters: parameters: