I find a BUG on loss_weight.

I find loss weight become biger when time step is smaller. But flow matching method usually train more on big time step.  is it a bug or you design it？

shift as 3. Red point's time index is 900,  means timestep value is small because timestep is inverse sorted.
<img width="567" height="434" alt="Image" src="https://github.com/user-attachments/assets/74175283-c0b9-4baa-bfb1-f1dbf799b593" />