diff --git a/tests/utils.py b/tests/utils.py index 16a6a19f..7465b396 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -159,8 +159,6 @@ def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50): acc = torch.mean(acc).item() else: - import pdb - pdb.set_trace() y_hat = trained_model(x) # acc