This commit is contained in:
James Betker 2022-01-01 14:05:33 -07:00
parent dc535b5358
commit d8111e0477

View File

@ -90,7 +90,8 @@ class ResBlock(nn.Module):
class AudioMiniEncoder(nn.Module): class AudioMiniEncoder(nn.Module):
def __init__(self, spec_dim, def __init__(self,
spec_dim,
embedding_dim, embedding_dim,
base_channels=128, base_channels=128,
depth=2, depth=2,