Delete test.py
Browse files
test.py
DELETED
@@ -1,18 +0,0 @@
|
|
1 |
-
import torchaudio
|
2 |
-
import torch
|
3 |
-
import torch.nn as nn
|
4 |
-
import numpy as np
|
5 |
-
from functions import *
|
6 |
-
from unet import UNet
|
7 |
-
|
8 |
-
"""
|
9 |
-
model = UNet()
|
10 |
-
model_state_dict = torch.load("huggingface/model.pth", map_location="cpu")
|
11 |
-
model.load_state_dict(model_state_dict["model_state_dict"])
|
12 |
-
print("# of trainable parameters =", sum(p.numel() for p in model.parameters() if p.requires_grad))
|
13 |
-
|
14 |
-
audio = load_audio("huggingface/p232_001.wav")
|
15 |
-
enhanced = predict("huggingface/p232_001.wav", model)
|
16 |
-
print(enhanced.shape)"""
|
17 |
-
string = "C:/Users/durim/Documents/KTH/Master_År2/DT2119-SSR/project_feature_extraction.ipynb"
|
18 |
-
print("/".join(string.split("/")[:-1]))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|