Remove DeMemWM model debugger traps

algorithms/dememwm/models/diffusion.py

@@ -166,10 +166,7 @@ class Diffusion(nn.Module):
         x = x.permute(1,0,2,3,4)
         action_cond = action_cond.permute(1,0,2)
         if pose_cond is not None and pose_cond[0] is not None:
-            try:
-                pose_cond = pose_cond.permute(1,0,2)
-            except:
-                pass
+            pose_cond = pose_cond.permute(1,0,2)
         t = t.permute(1,0)
         if frame_memory_segments is not None and frame_memory_pose is not None:
             # Algorithm preprocessing keeps packed metadata as T_all x B x 5;

algorithms/dememwm/models/dit.py

@@ -677,10 +677,7 @@ class DiT(nn.Module):
         c = rearrange(c, "(b t) d -> b t d", t=T)
 
         if torch.is_tensor(action_cond):
-            try:
-                c_action_cond = c + self.external_cond(action_cond)
-            except:
-                import pdb;pdb.set_trace()
+            c_action_cond = c + self.external_cond(action_cond)
         else:
             c_action_cond = None