use checkpoint lock alone

zooniverse · Dec 2, 2024 · 952eeb7 · 952eeb7
1 parent d08d649
commit 952eeb7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/azure/batch/scripts/train_model_finetune_on_catalog.py b/azure/batch/scripts/train_model_finetune_on_catalog.py
@@ -106,8 +106,7 @@
     model = finetune.FinetuneableZoobotTree(
         # params specific to tree finetuning
         schema=schema,
-        zoobot_checkpoint_loc=args.checkpoint,
-        name='hf_hub:mwalmsley/zoobot-encoder-convnext_nano' #to be removed
+        zoobot_checkpoint_loc=args.checkpoint
     )
 
     trainer = finetune.get_trainer(