tckm.yan/aha
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update fmt

Browse Source
This commit is contained in:
jhqxxx
2026-04-17 15:10:00 +08:00
parent 1cf075fd1d
commit d91b9767fe
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/models/fire_red_vad/model.rs
+1 -3
View File
@@ -246,12 +246,10 @@ impl DFSMN {
let cache = caches.map(|caches| &caches[0]); let cache = caches.map(|caches| &caches[0]);
let (mut memory, mut new_cache) = self.fsmn1.forward(&p, mask.as_ref(), cache)?; let (mut memory, mut new_cache) = self.fsmn1.forward(&p, mask.as_ref(), cache)?;
new_caches.push(new_cache); new_caches.push(new_cache);
let mut i = 1; for (i, fsmn) in (1..).zip(self.fsmns.iter()) {
for fsmn in &self.fsmns {
let cache = caches.map(|caches| &caches[i]); let cache = caches.map(|caches| &caches[i]);
(memory, new_cache) = fsmn.forward(&memory, mask.as_ref(), cache)?; (memory, new_cache) = fsmn.forward(&memory, mask.as_ref(), cache)?;
new_caches.push(new_cache); new_caches.push(new_cache);
i += 1;
} }
for dnn in &self.dnns { for dnn in &self.dnns {
memory = dnn.forward(&memory)?.relu()?; memory = dnn.forward(&memory)?.relu()?;