Upload folder using huggingface_hub
Browse files- models/hparams.json +21 -0
- models/mask_sam_model_best.pth +3 -0
- models/mask_sam_model_latest.pth +3 -0
- models/train_log.csv +2 -0
models/hparams.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"num_epochs": 2,
|
| 3 |
+
"batch_size": 4,
|
| 4 |
+
"use_amp": false,
|
| 5 |
+
"use_wandb": false,
|
| 6 |
+
"use_boxes": true,
|
| 7 |
+
"use_masks": true,
|
| 8 |
+
"patience": 5,
|
| 9 |
+
"task_name": "mask_sam_auto_prompt_ViT-B",
|
| 10 |
+
"model_save_path": "./models",
|
| 11 |
+
"save_dir": "./epoch_outputs",
|
| 12 |
+
"num_saved": 7,
|
| 13 |
+
"good_ex": [
|
| 14 |
+
20,
|
| 15 |
+
267,
|
| 16 |
+
511
|
| 17 |
+
],
|
| 18 |
+
"use_mask_aux_loss": true,
|
| 19 |
+
"use_mask_final_loss": true,
|
| 20 |
+
"use_box_loss": true
|
| 21 |
+
}
|
models/mask_sam_model_best.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:58d7573c9f1f9d9ca9bb00bf0ca68afcc8f16f8e683783181f12a6e4ae5a5612
|
| 3 |
+
size 383547170
|
models/mask_sam_model_latest.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:130387d1eefa5c3a912e827bcb6790ff9bc307583bd4115a8554714c7cba19e1
|
| 3 |
+
size 383548202
|
models/train_log.csv
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datetime,epoch,train_loss,train_aux,train_final,train_box,val_loss,val_aux,val_final,val_box,val_dice
|
| 2 |
+
20250928-0047,0,2.6306951117515562,0.5909306885302067,0.8390757858753204,1.200688619017601,2.5689780044555666,0.5455476856231689,0.8227166247367859,1.2007136535644531,0.206406577359885
|