add model
Browse files
model.py
CHANGED
|
@@ -788,10 +788,3 @@ class SpectraAASIST(nn.Module, PyTorchModelHubMixin):
|
|
| 788 |
return x.item()
|
| 789 |
|
| 790 |
spectra_aasist = SpectraAASIST
|
| 791 |
-
if __name__ == "__main__":
|
| 792 |
-
model = SpectraAASIST()
|
| 793 |
-
model.load_state_dict(torch.load("/data/home/maslov/maslov/WEIGHTS/aasist/baseline_v1_linear/weights/model.pt", map_location="cpu"))
|
| 794 |
-
model.eval()
|
| 795 |
-
x = torch.randn(1, 64400)
|
| 796 |
-
print(model(x).shape)
|
| 797 |
-
print(model(x))
|
|
|
|
| 788 |
return x.item()
|
| 789 |
|
| 790 |
spectra_aasist = SpectraAASIST
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|