[O] Use relative imports, remove unused imports

2023-10-11 18:56:18 -04:00
parent ea5629de26
commit eeaa2a3a34
5 changed files with 7 additions and 18 deletions
@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from torchlibrosa.stft import Spectrogram, LogmelFilterBank
-from models.htsat import HTSATWrapper
+from .htsat import HTSATWrapper

 def get_audio_encoder(name: str):
    if name == "Cnn14":