Glen9010

maybe you can try again, the code has updated 3 months ago. Beside, there still exit memory leakage In ComplexBatchNorm1d. You can add "with torch.no_grad():" after code line 254