From e3a7194d59565b806785fdce03bf7237b3329fd9 Mon Sep 17 00:00:00 2001 From: yangdongchao <15087581161@163.com> Date: Thu, 6 Apr 2023 22:18:47 +0800 Subject: [PATCH] fix e --- audio_detection/target_sound_detection/src/models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/audio_detection/target_sound_detection/src/models.py b/audio_detection/target_sound_detection/src/models.py index eeeec40..0af9d01 100644 --- a/audio_detection/target_sound_detection/src/models.py +++ b/audio_detection/target_sound_detection/src/models.py @@ -1113,9 +1113,9 @@ class RaDur_fusion(nn.Module): self.detection = CDur_CNN_mul_scale_fusion(inputdim, outputdim, time_resolution) self.softmax = nn.Softmax(dim=2) #self.temperature = 5 - if model_config['pre_train']: - self.encoder.load_state_dict(torch.load(model_config['encoder_path'])['model']) - self.detection.load_state_dict(torch.load(model_config['CDur_path'])) + # if model_config['pre_train']: + # self.encoder.load_state_dict(torch.load(model_config['encoder_path'])['model']) + # self.detection.load_state_dict(torch.load(model_config['CDur_path'])) self.q = nn.Linear(128,128) self.k = nn.Linear(128,128)