Skip to content

Commit c94ceba

Browse files
Update audiocaps example config
1 parent f9d3aac commit c94ceba

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

configs/audiocaps/waveform/cnn14rnn_trm.yaml

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
inherit_from: ./train_data.yaml
22

3-
remark: cola_cnn14rnn_trm
3+
remark: cnn14rnn_trm
44

55
# sync_file: /mnt/lustre/sjtu/home/xnx98/work/AudioCaption/ddp_sync/sync
66

@@ -9,8 +9,7 @@ model:
99
type: Cnn14RnnEncoder
1010
args:
1111
sample_rate: 32000
12-
# pretrained: sed/audioset_tagging_cnn/pretrained_weights/Cnn14_mAP=0.431.pth
13-
pretrained: /mnt/lustre/sjtu/home/xnx98/work/Cola/experiments/audioset_cnn14/best.pth
12+
pretrained: sed/audioset_tagging_cnn/pretrained_weights/Cnn14_mAP=0.431.pth
1413
freeze_cnn: True
1514
freeze_cnn_bn: True
1615
bidirectional: True
@@ -43,9 +42,6 @@ lr_scheduler:
4342
# warmup_iters: 3000
4443
final_lrs: !!float 5e-7
4544

46-
# resume: experiments/audioset/label_any/TransformerModel/cnn14rnn_trm/retrieve_model_iteration_2/expdecay_scheduler/seed_1/best.pth
47-
48-
max_aug_discount: 1.0
4945

5046
trainer:
5147
max_grad_norm: 1.0

0 commit comments

Comments
 (0)