update_save_order

This commit is contained in:
Yizhou Chi 2024-09-27 12:54:22 +08:00
parent 2b67355358
commit 24536df24c

View file

@ -31,9 +31,6 @@ class MCTSExperimenter(Experimenter):
dev_best_node = best_nodes["dev_best"]
score_dict = best_nodes["scores"]
self.copy_notebook(best_node, "best")
self.copy_notebook(dev_best_node, "dev_best")
text, num_generated_codes = get_tree_text(mcts.root_node)
text += f"Generated {num_generated_codes} unique codes.\n"
text += f"Best node: {best_node.id}, score: {best_node.raw_reward}\n"
@ -55,6 +52,8 @@ class MCTSExperimenter(Experimenter):
}
]
self.save_result(results)
self.copy_notebook(best_node, "best")
self.copy_notebook(dev_best_node, "dev_best")
def copy_notebook(self, node, name):
node_dir = node.get_node_dir()