• Thorsten Kurth's avatar
    fixing amp stuff again (#48) · f1a965bd
    Thorsten Kurth authored
    - doing manual type conversion in custom autograd
    - fixing stride issues in backward pass by making some output tensors contiguous
    f1a965bd
convolution.py 22.5 KB