fix(): clean code

brdhunga · Oct 6, 2021 · 57dbf7d · 57dbf7d
1 parent ec16745
commit 57dbf7d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/feature_layers.py b/feature_layers.py
@@ -1,8 +1,6 @@
 import torch.nn as nn
 import torch
-import torch.nn.functional as F
 import tools.audio_utils
-from scipy.fft import dct
 import librosa
 import numpy as np
 
@@ -46,6 +44,7 @@ def forward(self, x, device):
 
         return batch_output.to(device)
 
+
 class LinearDCT(nn.Linear):
     """Implement any DCT as a linear layer; in practice this executes around
     50x faster on GPU. Unfortunately, the DCT matrix is stored, which will

diff --git a/tools/dataset_loader.py b/tools/dataset_loader.py
@@ -34,11 +34,6 @@ def __init__(self,
 
         self.protocols_fname = os.path.join(self.data_root, self.dset_name + '.protocol.txt')
 
-        #         if is_eval:
-        #             data_root = os.path.join('eval', data_root)
-        #         if is_eval2021:
-        #             data_root = os.path.join('eval2021', data_root)
-
         self.files_dir = os.path.join(self.data_root, '{}'.format(self.dset_name))
         self.transform = transform