search

wavefrontshaping / complexPyTorch

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

Fix args for ComplexAvgPool2d #18

Closed jackhwalters closed 11 months ago

jackhwalters commented 2 years ago

Need to replace dilation and return_indices args with count_include_pad and divisor_override args to be compatible with torch.nn.functional.avg_pool2d

aeeeeeep commented 1 year ago

Good Job!