causality

[saeeddhqan]

flash-stu/flash_stu/utils/stu_utils.py

Line 55 in 54f78d4

U = torch.fft.rfft(U, n=n, dim=1)

Are you sure that the model is causal?