Change the dims[0] and dims[1] positioning for Spectrogram

SundarRajan28 · Mar 25, 2024 · d791b9a · d791b9a
1 parent cb10bdc
commit d791b9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rocAL/source/api/rocal_api_augmentation.cpp b/rocAL/source/api/rocal_api_augmentation.cpp
@@ -2218,7 +2218,7 @@ rocalSpectrogram(
         max_frame = std::max(0, max_frame);
         int bins = std::max(0, (nfft / 2) + 1);
         std::vector<size_t> dims = output_info.dims();
-        if (spectrogram_layout == RocalSpectrogramLayout::ROCAL_FT) {
+        if (spectrogram_layout == RocalSpectrogramLayout::ROCAL_TF) {
             dims[1] = max_frame;
             dims[2] = bins;
         } else {