Merge branch 'main' of github.com:sisl/InteractionImitation

2021-08-02 11:08:35 +02:00
parent 281f7773c4 fb91ee1a62
commit 4ea4d42df7
4 changed files with 45 additions and 30 deletions
--- a/src/metrics.py
+++ b/src/metrics.py
@@ -58,8 +58,6 @@ def visualize_distribution(true, pred, filestr):
    """
    nni1 = ~torch.isnan(true)
    nni2 = ~torch.isnan(pred)
-    import pdb
-    pdb.set_trace()
    plt.figure()
    plt.hist(true[nni1].numpy(), density=True, bins=20)
    plt.hist(pred[nni2].numpy(), density=True, bins=20)