replace check for empty tensor.

Abrikosoff · Feb 13, 2019 · 714b6c4 · 714b6c4
1 parent a344d75
commit 714b6c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchdiffeq/_impl/adjoint.py b/torchdiffeq/_impl/adjoint.py
@@ -75,7 +75,7 @@ def augmented_dynamics(t, y_aug):
                 time_vjps.append(dLd_cur_t)
 
                 # Run the augmented system backwards in time.
-                if len(adj_params) == 0:
+                if adj_params.numel() == 0:
                     adj_params = torch.tensor(0.).to(adj_y[0])
                 aug_y0 = (*ans_i, *adj_y, adj_time, adj_params)
                 aug_ans = odeint(