修改cnn为合理的包名

夜雨飘零

2023-12-19 53fccccb24d15d788919d91c8c2b06a115ddacf3

修改cnn为合理的包名

 funasr/models/campplus/__init__.py

@@ -0,0 +1 @@
from .campplus import CAMPPlus

 funasr/models/campplus/campplus.py

File was renamed from funasr/models/cnn/DTDNN.py
@@ -6,7 +6,7 @@
import torch.nn.functional as F
from torch import nn

from funasr.modules.cnn.layers import DenseLayer, StatsPool, TDNNLayer, CAMDenseTDNNBlock, TransitLayer, \
from funasr.models.campplus.layers import DenseLayer, StatsPool, TDNNLayer, CAMDenseTDNNBlock, TransitLayer, \
    BasicResBlock, get_nonlinear



 funasr/models/campplus/layers.py

 funasr/models/cnn/__init__.py

File was deleted

 funasr/models/eres2net/__init__.py

New file
@@ -0,0 +1,2 @@
from .eres2net import ERes2Net
from .eres2net_aug import ERes2NetAug

 funasr/models/eres2net/eres2net.py

File was renamed from funasr/models/cnn/ResNet.py
@@ -16,7 +16,7 @@
import torch.nn.functional as F

import funasr.models.sond.pooling.pooling_layers as pooling_layers
from funasr.modules.cnn.fusion import AFF
from funasr.models.eres2net.fusion import AFF


class ReLU(nn.Hardtanh):

 funasr/models/eres2net/eres2net_aug.py

File was renamed from funasr/models/cnn/ResNet_aug.py
@@ -16,7 +16,7 @@
import torch.nn.functional as F

import funasr.models.sond.pooling.pooling_layers as pooling_layers
from funasr.modules.cnn.fusion import AFF
from funasr.models.eres2net.fusion import AFF


class ReLU(nn.Hardtanh):

 funasr/models/eres2net/fusion.py