search

wavefrontshaping / complexPyTorch

A high-level toolbox for using complex valued neural networks in PyTorch
MIT License
610 stars 148 forks source link

fix memory leak + reformatting #23

Closed kareem-metwaly closed 11 months ago

kareem-metwaly commented 2 years ago

This PR fixes the memory leak with ComplexBatchNorm1d (I added with torch.no_grad as it was missing)

kareem-metwaly commented 2 years ago

There were some merge conflicts that I have fixed