cybertronai / autograd-hacks

The Unlicense
153 stars 32 forks source link

Support of padding, dilation and stride in compute_grad1 function #3

Open oljike opened 4 years ago

oljike commented 4 years ago

Propose to substitute the following line: https://github.com/cybertronai/autograd-hacks/blob/1fa974aa32a50aeadec449147acc5ae5a032f848/autograd_hacks.py#L165

with: A = torch.nn.functional.unfold(A, layer.kernel_size, dilation=layer.dilation, padding=layer.padding, stride=layer.stride)