maes_microscopy icon indicating copy to clipboard operation
maes_microscopy copied to clipboard

Update huggingface_mae.py

Open fatwir opened this issue 6 months ago • 0 comments

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.

fatwir avatar Aug 21 '25 22:08 fatwir