diff --git a/src/main.py b/src/main.py index 3e93d91..71eba85 100644 --- a/src/main.py +++ b/src/main.py @@ -181,7 +181,7 @@ def weight_stats(model): filename = f"update-{args.update}-lr-{args.lr}-wd-{args.wd}-seed-{args.seed}.pkl" os.makedirs("results", exist_ok=True) - print(f"Saving results to {os.path.join("results", filename)}") + print(f"Saving results to {os.path.join('results', filename)}") with open(os.path.join("results", filename), "wb") as f: pickle.dump(results, f) - print(f"Results saved to {os.path.join("results", filename)}") + print(f"Results saved to {os.path.join('results', filename)}")