Upload 3 files

2024-07-31T221951_CellViT-unireplknet-fold1/checkpoints/latest_checkpoint.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5c8f1deeb2953fb58224b8106a0e774db15640b48bc8c821f76a61ef5231dbb
+size 1515856855

2024-07-31T221951_CellViT-unireplknet-fold1/checkpoints/model_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b632ae5d6e23b6d22b42ff2885ff8517aa6646102a1c9d33567c507166915cd
+size 1515856855

2024-07-31T221951_CellViT-unireplknet-fold1/config.yaml

@@ -0,0 +1,156 @@
+logging:
+  log_dir: /data/ziweicui/cellvit-unireplknet-n/2024-07-31T221951_CellViT-unireplknet-fold1
+  mode: online
+  project: Cell-Segmentation
+  notes: CellViT-256
+  log_comment: CellViT-unireplknet-fold1
+  tags:
+  - Fold-1
+  - ViT256
+  wandb_dir: /home/ziweicui/cell_unireplknet/results
+  level: Debug
+  group: CellViT256
+  run_id: plcdxlzv
+  wandb_file: plcdxlzv
+random_seed: 19
+gpu: 0
+data:
+  dataset: PanNuke
+  dataset_path: /data/ziweicui/cellvit-png
+  train_folds:
+  - 1
+  val_folds:
+  - 0
+  test_folds:
+  - 2
+  num_nuclei_classes: 6
+  num_tissue_classes: 19
+model:
+  backbone: default
+  pretrained_encoder: /data/ziweicui/cellvit-uninet-1k-s/unireplknet_s_in1k_224_acc83.91.pth
+  shared_skip_connections: true
+loss:
+  nuclei_binary_map:
+    focaltverskyloss:
+      loss_fn: FocalTverskyLoss
+      weight: 1
+    dice:
+      loss_fn: dice_loss
+      weight: 1
+  hv_map:
+    mse:
+      loss_fn: mse_loss_maps
+      weight: 2.5
+    msge:
+      loss_fn: msge_loss_maps
+      weight: 8
+  nuclei_type_map:
+    bce:
+      loss_fn: xentropy_loss
+      weight: 0.5
+    dice:
+      loss_fn: dice_loss
+      weight: 0.2
+    mcfocaltverskyloss:
+      loss_fn: MCFocalTverskyLoss
+      weight: 0.5
+      args:
+        num_classes: 6
+  tissue_types:
+    ce:
+      loss_fn: CrossEntropyLoss
+      weight: 0.1
+training:
+  drop_rate: 0
+  attn_drop_rate: 0.1
+  drop_path_rate: 0.1
+  batch_size: 16
+  world_size: 2
+  rank: 0
+  epochs: 100
+  optimizer: AdamW
+  early_stopping_patience: 70
+  scheduler:
+    scheduler_type: cosine
+    hyperparameters:
+      eta_min: 1e-5
+  optimizer_hyperparameter:
+    betas:
+    - 0.85
+    - 0.95
+    lr: 0.001
+    weight_decay: 0.1
+    opt_eps: 8.0e-08
+  layer_decay: 0.999999
+  unfreeze_epoch: 0
+  sampling_gamma: 0.85
+  sampling_strategy: cell+tissue
+  mixed_precision: false
+transformations:
+  randomrotate90:
+    p: 0.5
+  horizontalflip:
+    p: 0.5
+  verticalflip:
+    p: 0.5
+  downscale:
+    p: 0.15
+    scale: 0.5
+  blur:
+    p: 0.2
+    blur_limit: 10
+  gaussnoise:
+    p: 0.25
+    var_limit: 50
+  colorjitter:
+    p: 0.2
+    scale_setting: 0.25
+    scale_color: 0.1
+  superpixels:
+    p: 0.1
+  zoomblur:
+    p: 0.1
+  randomsizedcrop:
+    p: 0.1
+  elastictransform:
+    p: 0.2
+  normalize:
+    mean:
+    - 0.5
+    - 0.5
+    - 0.5
+    std:
+    - 0.5
+    - 0.5
+    - 0.5
+eval_checkpoint: latest_checkpoint.pth
+dataset_config:
+  tissue_types:
+    Adrenal_gland: 0
+    Bile-duct: 1
+    Bladder: 2
+    Breast: 3
+    Cervix: 4
+    Colon: 5
+    Esophagus: 6
+    HeadNeck: 7
+    Kidney: 8
+    Liver: 9
+    Lung: 10
+    Ovarian: 11
+    Pancreatic: 12
+    Prostate: 13
+    Skin: 14
+    Stomach: 15
+    Testis: 16
+    Thyroid: 17
+    Uterus: 18
+  nuclei_types:
+    Background: 0
+    Neoplastic: 1
+    Inflammatory: 2
+    Connective: 3
+    Dead: 4
+    Epithelial: 5
+run_sweep: false
+agent: null