Spaces:

aadarsh99
/

ConverSeg

Running on Zero

aadarsh99 commited on Jan 16

Commit

a7f2ace

1 Parent(s): 6221bbd

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,13 +36,13 @@ MODEL_CONFIGS = {
     },
     "Stage 2 (grad-acc: 4)": {
         "repo_id": "aadarsh99/ConvSeg-Stage2",
-        "sam_filename": "fine_tuned_sam2_batched_60000.torch",
-        "plm_filename": "fine_tuned_sam2_batched_plm_60000.torch"
     },
     "Stage 2 (grad-acc: 8)": {
         "repo_id": "aadarsh99/ConvSeg-Stage2",
-        "sam_filename": "fine_tuned_sam2_batched_100000.torch",
-        "plm_filename": "fine_tuned_sam2_batched_plm_100000.torch"
     }
 }

     },
     "Stage 2 (grad-acc: 4)": {
         "repo_id": "aadarsh99/ConvSeg-Stage2",
+        "sam_filename": "fine_tuned_sam2_batched_18000.torch",
+        "plm_filename": "fine_tuned_sam2_batched_plm_18000.torch"
     },
     "Stage 2 (grad-acc: 8)": {
         "repo_id": "aadarsh99/ConvSeg-Stage2",
+        "sam_filename": "fine_tuned_sam2_batched_18000.torch",
+        "plm_filename": "fine_tuned_sam2_batched_plm_18000.torch"
     }
 }