updated with latest PL and Ray (#15653)

2022-02-16 04:53:05 +13:00
parent 7bc4a01cb5
commit 80f1a59168
4 changed files with 21 additions and 9 deletions
--- a/examples/research_projects/rag/callbacks_rag.py
+++ b/examples/research_projects/rag/callbacks_rag.py
@@ -38,7 +38,7 @@ def get_checkpoint_callback(output_dir, metric):
        monitor=f"val_{metric}",
        mode="max",
        save_top_k=3,
-        period=1,  # maybe save a checkpoint every time val is run, not just end of epoch.
+        every_n_epochs=1,  # maybe save a checkpoint every time val is run, not just end of epoch.
    )
    return checkpoint_callback