Skip to content

Commit c3bee7f

Browse files
committed
removed unnecessary padding in sv model input
1 parent 7fe459e commit c3bee7f

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

train.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -303,12 +303,6 @@ def train_one_step(self, batch):
303303
sample_frequency=16000)
304304
feat = feat - feat.mean(dim=0, keepdim=True)
305305
feat_list.append(feat)
306-
max_feat_len = max([feat.size(0) for feat in feat_list])
307-
feat_lens = torch.tensor([feat.size(0) for feat in feat_list], dtype=torch.int32).to(self.device) // 2
308-
feat_list = [
309-
torch.nn.functional.pad(feat, (0, 0, 0, max_feat_len - feat.size(0)), value=float(feat.min().item()))
310-
for feat in feat_list
311-
]
312306
y_list = []
313307
with torch.no_grad():
314308
for feat in feat_list:

0 commit comments

Comments
 (0)