minor fixes in original RAG training (#12395)

2021-06-30 00:39:48 +12:00
parent e3f39a2952
commit 5257818e68
2 changed files with 3 additions and 3 deletions
--- a/examples/research_projects/rag/callbacks_rag.py
+++ b/examples/research_projects/rag/callbacks_rag.py
@@ -36,7 +36,7 @@ def get_checkpoint_callback(output_dir, metric):
        dirpath=output_dir,
        filename=exp,
        monitor=f"val_{metric}",
-        mode="min",
+        mode="max",
        save_top_k=3,
        period=1,  # maybe save a checkpoint every time val is run, not just end of epoch.
    )