diff --git a/tests/debug.py b/tests/debug.py index 52bd0ec56f..cb6306ed48 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -91,7 +91,7 @@ def run_prediction(dataloader, trained_model): assert val_acc > 0.70, f'this model is expected to get > 0.7 in test set (it got {val_acc})' -def main(): +def mainasdf(): save_dir = init_save_dir() model, hparams = get_model() @@ -145,5 +145,6 @@ def get_pids(port): except: pass + if __name__ == '__main__': get_pids(12910) \ No newline at end of file