search

XinhaoMei / DCASE2021_task6_v2

Code for CVSSP submission to DCASE 2021 Task 6
35 stars 5 forks source link

About the reinforcement learning #1

Closed LittleFlyingSheep closed 3 years ago

LittleFlyingSheep commented 3 years ago

Hi, thank you very much for your work! I am reading your code with report to understand the work you did. How do you implement the reinforcement learning in "config.mode == 'finetune' "? Specifically, which part of the code is the 'CIDEr loss' part in 'Fine-tuning using RL'?

XinhaoMei commented 3 years ago

Hi, thank you very much for your work! I am reading your code with report to understand the work you did. How do you implement the reinforcement learning in "config.mode == 'finetune' "? Specifically, which part of the code is the 'CIDEr loss' part in 'Fine-tuning using RL'?

Hi, the code for reinforcement learning is in finetune_rl.py. And you should set the corresponding settings in rl block in the setting file. Hope this helpful. Thanks!

LittleFlyingSheep commented 3 years ago

Thank you very much! It is my inattention that I only notice the 'train.py' but ignore the 'finetune.py'.