maes_microscopy
maes_microscopy copied to clipboard
Update huggingface_mae.py
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.