support funasr 1.0 (#1346)

夜雨飘零

2024-02-02 85c08383831ea2b7cdf4c6f863f71b20b95b6782

 funasr/models/specaug/profileaug.py

@@ -2,10 +2,9 @@
import numpy as np
import torch
from torch.nn import functional as F
from funasr.models.specaug.abs_profileaug import AbsProfileAug
import torch.nn as nn


class ProfileAug(AbsProfileAug):
class ProfileAug(nn.Module):
    """
    Implement the augmentation for profiles including:
    - Split aug: split one profile into two profiles, i.e., main and inaccurate, labels assigned to main