Mirrors/AudioGPT
1
0
Fork 0
mirror of https://github.com/AIGC-Audio/AudioGPT.git synced 2025-12-15 19:37:52 +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.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)