Skip to content
Snippets Groups Projects
Commit 62893567 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Lint

parent 0f65277f
No related merge requests found
Pipeline #22365 failed with stage
in 1 minute and 29 seconds
...@@ -64,7 +64,7 @@ def create_autograd_function(autodiff_obj, use_cuda): ...@@ -64,7 +64,7 @@ def create_autograd_function(autodiff_obj, use_cuda):
field.shape, field.shape,
dtype=numpy_dtype_to_torch(field.dtype.numpy_dtype), dtype=numpy_dtype_to_torch(field.dtype.numpy_dtype),
device=next(chain(args, kwargs.values())).device) device=next(chain(args, kwargs.values())).device)
except: except Exception:
shape = next(filter(lambda x: isinstance(x, torch.Tensor), chain(args, kwargs.values()))).shape shape = next(filter(lambda x: isinstance(x, torch.Tensor), chain(args, kwargs.values()))).shape
kwargs[field.name] = torch.zeros( kwargs[field.name] = torch.zeros(
shape, shape,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment