From 952eeb791b96a00be356757b1616e7eee07bbe8c Mon Sep 17 00:00:00 2001
From: tooyosi <oluwatoyosi.oyegoke@gmail.com>
Date: Mon, 2 Dec 2024 21:16:25 +0000
Subject: [PATCH] use checkpoint lock alone

---
 azure/batch/scripts/train_model_finetune_on_catalog.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/azure/batch/scripts/train_model_finetune_on_catalog.py b/azure/batch/scripts/train_model_finetune_on_catalog.py
index 34a3b17..3c80a0f 100644
--- 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(