add config file for lightning checkpoint

SalesforceAIResearch · Aug 8, 2024 · ea6b44e · ea6b44e
1 parent fac7a93
commit ea6b44e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cli/conf/eval/model/moirai_lightning_ckpt.yaml b/cli/conf/eval/model/moirai_lightning_ckpt.yaml
@@ -0,0 +1,5 @@
+_target_: uni2ts.model.moirai.MoiraiForecast.load_from_checkpoint
+checkpoint_path: ...
+num_samples: 100
+patch_size: ???
+context_length: ???