From 9595a9432fadfbdacd4e6897f6b9a83957699558 Mon Sep 17 00:00:00 2001
From: seanzhang-zhichen <74812416+seanzhang-zhichen@users.noreply.github.com>
Date: 星期二, 05 三月 2024 17:42:14 +0800
Subject: [PATCH] modify paraformer train doc (#1427)
---
funasr/models/emotion2vec/timm_modules.py | 12 ++++--------
1 files changed, 4 insertions(+), 8 deletions(-)
diff --git a/funasr/models/emotion2vec/timm_modules.py b/funasr/models/emotion2vec/timm_modules.py
index 1f6285a..b26da52 100644
--- a/funasr/models/emotion2vec/timm_modules.py
+++ b/funasr/models/emotion2vec/timm_modules.py
@@ -1,12 +1,8 @@
-from itertools import repeat
-import collections.abc
-from functools import partial
-from typing import Optional, Tuple
-import numpy as np
-
-import torch
import torch.nn as nn
-import torch.nn.functional as F
+import collections.abc
+from itertools import repeat
+from functools import partial
+
def drop_path(x, drop_prob: float = 0., training: bool = False, scale_by_keep: bool = True):
"""Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
--
Gitblit v1.9.1