Merge pull request #17 from mwalmsley/refactor-pytorch-datasets

missed a rename, thanks cam
mwalmsley · Apr 29, 2022 · 0893114 · 0893114
2 parents 4567825 + deff1f3
commit 0893114
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zoobot/pytorch/examples/train_model_on_catalog.py b/zoobot/pytorch/examples/train_model_on_catalog.py
@@ -79,7 +79,7 @@
     else:
         wandb_logger = None
 
-    train_with_pytorch_lightning.train(
+    train_with_pytorch_lightning.train_default_zoobot_from_scratch(
         save_dir=args.save_dir,
         catalog=catalog,
         schema=schema,