misc
This commit is contained in:
parent
dc535b5358
commit
d8111e0477
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue
Block a user