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

Lint

parent 0f65277f
Branches
Tags
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):
field.shape,
dtype=numpy_dtype_to_torch(field.dtype.numpy_dtype),
device=next(chain(args, kwargs.values())).device)
except:
except Exception:
shape = next(filter(lambda x: isinstance(x, torch.Tensor), chain(args, kwargs.values()))).shape
kwargs[field.name] = torch.zeros(
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