Mirrors/AudioGPT
1
0
Fork 0
mirror of https://github.com/AIGC-Audio/AudioGPT.git synced 2025-12-16 11:57:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix e

Browse Source
This commit is contained in:
yangdongchao
2023-04-06 22:18:47 +08:00
parent 322ed8cbb2
commit e3a7194d59
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
audio_detection/target_sound_detection/src/models.py
View File

@@ -1113,9 +1113,9 @@ class RaDur_fusion(nn.Module):
self.detection = CDur_CNN_mul_scale_fusion(inputdim, outputdim, time_resolution) self.detection = CDur_CNN_mul_scale_fusion(inputdim, outputdim, time_resolution)
self.softmax = nn.Softmax(dim=2) self.softmax = nn.Softmax(dim=2)
#self.temperature = 5 #self.temperature = 5
if model_config['pre_train']: # if model_config['pre_train']:
self.encoder.load_state_dict(torch.load(model_config['encoder_path'])['model']) # self.encoder.load_state_dict(torch.load(model_config['encoder_path'])['model'])
self.detection.load_state_dict(torch.load(model_config['CDur_path'])) # self.detection.load_state_dict(torch.load(model_config['CDur_path']))
self.q = nn.Linear(128,128) self.q = nn.Linear(128,128)
self.k = nn.Linear(128,128) self.k = nn.Linear(128,128)