• Maciej Torhan's avatar
    Fix Adam and AdamW initializers in wav2letter example (#3145) · cea12eaf
    Maciej Torhan authored
    Summary:
    In wav2letter example there is passed `momentum` to `Adam` and `AdamW` initializer, which is not a correct parameter. To fix that we need to add `beta_1` and `beta_2` to arguments and replace `momentum` with them. I also added `eps` similar to `Adadelta` initializer.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/3145
    
    Reviewed By: mthrok
    
    Differential Revision: D43847713
    
    Pulled By: nateanl
    
    fbshipit-source-id: 94f7c48232fabf520cfce81471694cb545d160c6
    cea12eaf
main.py 18.4 KB