Fatwir Sheikh Mohammed
Results
2
issues of
Fatwir Sheikh Mohammed
## Proposed Change Adjust Fourier loss calculation for different dimensions. Currently, the loss dimensions expected are 2, but applying `mean(dim=-1)` on 4D inputs leads to issues.
There is a small typo, and `self.num_bins` should be replaced with `self.num_modalities`