Don't load dataset during inference

1 year ago · 131dbf4b5b
parent e4b6069e10
commit 131dbf4b5b
1 changed files with 1 additions and 1 deletions
--- a/src/python/larynx_train/infer.py
+++ b/src/python/larynx_train/infer.py
@ -29,7 +29,7 @@ def main():
    args.output_dir = Path(args.output_dir)
    args.output_dir.mkdir(parents=True, exist_ok=True)

-    model = VitsModel.load_from_checkpoint(args.checkpoint)
+    model = VitsModel.load_from_checkpoint(args.checkpoint, dataset=None)

    # Inference only
    model.eval()