single-shot-robot

Josselin's group workspace

Group: 4-DICE

1-20

of 20

Tags

Notes

Author

josselin

State

Finished

Start time

September 8th, 2023 3:11:15 AM

Runtime

41m 6s

Tracked hours

41m 1s

Run path

single-shot-robot/SKIL-Reachbot_cracks_combined_segmentation_prod/328oh3i6

Linux-5.15.0-79-generic-x86_64-with-glibc2.17

Python version

3.8.16

Git repository

git clone git@github.com:JosselinSomervilleRoberts/ReachBot-Predictor.git

Git state

git checkout -b "4-DICE_R1RYTPEQ3W" 5b8a1208cc806c5bd7b959bb78d50ab2a583518d

Command

tools/train.py paper_configs/vit/cracks/dice.py --amp

System Hardware

CPU count	18
Logical CPU count	36
GPU count	1
GPU type	NVIDIA RTX A6000

W&B CLI Version

0.15.4

Group

4-DICE

Config parameters are your model's inputs. Learn more

▶
Config parameters:{} 54 keys
- CHECKPOINT:
  "pretrain/upernet_vit-b16_mln_512x512_80k_ade20k_20210624_130547-0403cee1_fix.pth"
- ▶
  crop_size:[] 2 items
  - 0:
    512
  - 1:
    512
- ▶
  data_preprocessor:{} 7 keys
- data_root:
  "../datasets/cracks_combined/"
- DATASET:
  "cracks_combined"
- dataset_type:
  "ReachbotDataset"
- ▶
  datasets_mappings:{} 6 keys
- DEBUG_PATH:
  null
- DEBUG_STEP_INTERVAL:
  -1
- ▶
  default_checkpoint:{} 2 keys
  - vit_vit-b16_mln_upernet_cracks:
    "pretrain/upernet_vit-b16_mln_512x512_80k_ade20k_20210624_130547-0403cee1_fix.pth"
  - vit_vit-b16_mln_upernet_vessels:
    "pretrain/upernet_vit-b16_mln_512x512_80k_ade20k_20210624_130547-0403cee1_fix.pth"
- ▶
  default_hooks:{} 5 keys
- default_scope:
  "mmseg"
- ▶
  env_cfg:{} 3 keys
  - cudnn_benchmark:
    true
  - ▶
    dist_cfg:{} 1 key
    - backend:
      "nccl"
  - ▶
    mp_cfg:{} 2 keys
    - mp_start_method:
      "fork"
    - opencv_num_threads:
      0
- EVAL_EPOCH_INTERVAL:
  5
- launcher:
  "none"
- load_from:
  null
- log_level:
  "INFO"
- ▶
  log_processor:{} 1 key
  - by_epoch:
    false
- LOG_STEP_INTERVAL:
  10
- ▶
  LOSSES:{} 1 key
  - dice:
    4
- ▶
  losses_mappings:{} 5 keys
- LR:
  "aggressive"
- memory_available:
  44.70343017578125
- METRIC:
  "ReachbotMetric"
- MODEL:
  "vit_vit-b16_mln_upernet_cracks"
- ▶
  model:{} 9 keys
- ▶
  norm_cfg:{} 2 keys
  - requires_grad:
    true
  - type:
    "SyncBN"
- num_epochs:
  100
- num_training_samples:
  186
- ▶
  optim_wrapper:{} 4 keys
  - loss_scale:
    "dynamic"
  - ▶
    optimizer:{} 4 keys
    - ▶
      betas:[] 2 items
      - 0:
        0.9
      - 1:
        0.999
    - lr:
      0.00001
    - type:
      "AdamW"
    - weight_decay:
      0.01
  - ▶
    paramwise_cfg:{} 1 key
    - ▶
      custom_keys:{} 3 keys
      - ▶
        cls_token:{} 1 key
        decay_mult:
        0
      - ▶
        norm:{} 1 key
        decay_mult:
        0
      - ▶
        pos_embed:{} 1 key
        decay_mult:
        0
  - type:
    "AmpOptimWrapper"
- ▶
  param_scheduler:[] 2 items
  - ▶
    0:{} 5 keys
  - ▶
    1:{} 6 keys
- resume:
  false
- SAVE_EPOCH_INTERVAL:
  -1
- ▶
  schedules_mappings:{} 2 keys
  - 40k:
    "custom_configs.schedules.40k"
  - aggressive:
    "custom_configs.schedules.aggressive"
- tag:
  "R1RYTPEQ3W"
- ▶
  test_cfg:{} 1 key
  - type:
    "TestLoop"
- ▶
  test_dataloader:{} 5 keys
- ▶
  test_evaluator:{} 1 key
  - type:
    "ReachbotMetric"
- ▶
  test_pipeline:[] 4 items
  - ▶
    0:{} 1 key
    - type:
      "LoadImageFromFile"
  - ▶
    1:{} 3 keys
    - keep_ratio:
      true
    - ▶
      scale:[] 2 items
      - 0:
        1,024
      - 1:
        512
    - type:
      "Resize"
  - ▶
    2:{} 2 keys
    - reduce_zero_label:
      false
    - type:
      "LoadAnnotations"
  - ▶
    3:{} 1 key
    - type:
      "PackSegInputs"
- train_batch_size:
  4
- ▶
  train_cfg:{} 3 keys
  - max_iters:
    4,650
  - type:
    "IterBasedTrainLoop"
  - val_interval:
    232
- ▶
  train_dataloader:{} 5 keys
- ▶
  train_pipeline:[] 7 items
- ▶
  tta_model:{} 1 key
  - type:
    "SegTTAModel"
- USE_WANDB:
  true
- ▶
  val_cfg:{} 1 key
  - type:
    "ValLoop"
- ▶
  visualizer:{} 4 keys
  - alpha:
    0.6
  - name:
    "visualizer"
  - type:
    "SegLocalVisualizer"
  - ▶
    vis_backends:[] 2 items
    - ▶
      0:{} 1 key
      - type:
        "LocalVisBackend"
    - ▶
      1:{} 2 keys
      - ▶
        init_kwargs:{} 4 keys
        entity:
        "single-shot-robot"
        group:
        "4-DICE"
        name:
        "4-DICE_R1RYTPEQ3W"
        project:
        "cracks_combined_segmentation_prod"
      - type:
        "WandbVisBackend"
- WANDB_PROJECT_SUFFIX:
  "_prod"
- WANDB_RUN_NAME:
  "4-DICE"
- work_dir:
  "./work_dirs/dice"

Summary metrics are your model's outputs. Learn more

▶
Summary metrics:{} 21 keys
- aux.acc_seg:
  97.29332733154295
- aux.loss_dice:
  0.6034803450107574
- avg:
  0.5566750009771672
- data_time:
  0.11103863716125488
- decode.acc_seg:
  97.38362121582033
- decode.loss_dice:
  1.237289547920227
- dice:
  0.400286994205805
- iou:
  0.25818319145984214
- iter:
  4,650
- length:
  0.7408028388466298
- line_distance:
  0.5610119809581885
- loss:
  1.84076988697052
- lr:
  0
- memory:
  8,559
- time:
  0.532166576385498
- ▶
  val_9.png:{} 7 keys
- ▶
  val_17.png:{} 7 keys
- ▶
  val_26.png:{} 7 keys
- ▶
  val_35.png:{} 7 keys
- ▶
  val_44.png:{} 7 keys
- width:
  0.6259512216028444

This run consumed these artifacts as inputs. Learn more

job

job-git_github.com_JosselinSomervilleRoberts_ReachBot-Predictor.git_mmsegmentation_tools_train.py:v14

This run produced these artifacts as outputs. Total: 2. Learn more

code

source-cracks_combined_segmentation_prod-mmsegmentation_tools_train.py:v17

wandb-history

run-328oh3i6-history:v0