[RNN-T] Spectogram Augmentation omits last row/column - Githubissues

mlcommons / training

Reference implementations of MLPerf™ training benchmarks

https://mlcommons.org/en/groups/training

Apache License 2.0

1.6k stars 553 forks source link

[RNN-T] Spectogram Augmentation omits last row/column #492

Closed mwawrzos closed 3 years ago

mwawrzos commented 3 years ago

Issue reported by @qpjaada

When time and frequency masks are sampled, the position of the masks never fills the last column, row respectively: https://github.com/mlcommons/training/blob/master/rnn_speech_recognition/pytorch/common/data/features.py#L102 https://github.com/mlcommons/training/blob/master/rnn_speech_recognition/pytorch/common/data/features.py#L88