Update modeling_wavcoch.py
Browse files- modeling_wavcoch.py +1 -1
modeling_wavcoch.py
CHANGED
|
@@ -135,7 +135,7 @@ class CochleagramTransform:
|
|
| 135 |
################$$$$$$##################
|
| 136 |
|
| 137 |
|
| 138 |
-
class
|
| 139 |
def __init__(self, config):
|
| 140 |
super().__init__()
|
| 141 |
self.N = config.window_size
|
|
|
|
| 135 |
################$$$$$$##################
|
| 136 |
|
| 137 |
|
| 138 |
+
class WavCoch(nn.Module, PyTorchModelHubMixin):
|
| 139 |
def __init__(self, config):
|
| 140 |
super().__init__()
|
| 141 |
self.N = config.window_size
|