TSTNN は単一のステートメントをテストします

import torch
import torch.nn as nn
from scipy.io import wavfile
import numpy as np
from checkpoints_best_pesq import Checkpoint
# from checkpoints import Checkpoint
from torch.utils.data import DataLoader
from helper_funcs import snr, numParams
from eval_composite import eval_composite
# from AudioData_PCS import EvalDataset, EvalCollate, TestDataset, TestCollate
from AudioData_complex_sc import EvalDataset, EvalCollate, TestDataset, TestCollate
from UNet import Net
import librosa
import h5py
import os


os.environ['CUDA_VISIBLE_DEVICES'] = '2'

sr = 16000

test_noisyaudio_path = '/home3/dataset/speech_enhancement/voicebank_demand/noisy_testset_wav/noisy_testset_wav/p232_005.wav'
test_cleanaudio_path = '/home3/dataset/speech_enhancement/voicebank_demand/clean_testset_wav/clean_testset_wav/p232_005.wav'

audio_file_save = '/home3/weiwb/code/Multi-Channel-Model-master/Single_Channel_Model/audio_tes