added debugging util

tests/debug.py

@@ -8,6 +8,8 @@ import warnings
 import torch
 import os
 import shutil
+import pdb
+
 
 def get_model():
     # set up model with these hyperparams