single-shot-robot

Josselin's group workspace

Group: 3-DICE-1-CLDICE

1-10

of 10

Tags

Notes

Author

josselin

State

Finished

Start time

August 20th, 2023 11:02:02 AM

Runtime

31m 43s

Tracked hours

31m 36s

Run path

single-shot-robot/SKIL-Reachbot_vessels_combined_shifted_segmentation_prod/cz8uhzm9

Linux-5.15.0-78-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 "3-DICE-1-CLDICE_03FMQ2460N" e267849eb4b38f81725e4b31e32a6eb7b541340e

Command

tools/train.py config_vessels_4.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

3-DICE-1-CLDICE

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/vessels/combined_degraded_shifted"
- DATASET:
  "vessels_combined_shifted"
- dataset_type:
  "ReachbotDataset"
- ▶
  datasets_mappings:{} 9 keys
- DEBUG_PATH:
  "/media/jsomerviller/SSD2/output"
- DEBUG_STEP_INTERVAL:
  -1
- ▶
  default_checkpoint:{} 1 key
  - vit_vit-b16_mln_upernet:
    "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:{} 2 keys
  - cl_dice:
    1
  - dice:
    3
- ▶
  losses_mappings:{} 5 keys
- LR:
  "aggressive"
- memory_available:
  47.0899658203125
- METRIC:
  "ReachbotMetric"
- MODEL:
  "vit_vit-b16_mln_upernet"
- ▶
  model:{} 9 keys
- ▶
  norm_cfg:{} 2 keys
  - requires_grad:
    true
  - type:
    "SyncBN"
- num_epochs:
  200
- num_training_samples:
  85
- ▶
  optim_wrapper:{} 4 keys
  - loss_scale:
    "dynamic"
  - ▶
    optimizer:{} 4 keys
    - ▶
      betas:[] 2 items
      - 0:
        0.9
      - 1:
        0.999
    - lr:
      0.00005
    - 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:{} 1 key
  - aggressive:
    "custom_configs.schedules.aggressive"
- tag:
  "03FMQ2460N"
- ▶
  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:
  8
- ▶
  train_cfg:{} 3 keys
  - max_iters:
    2,125
  - type:
    "IterBasedTrainLoop"
  - val_interval:
    53
- ▶
  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:
        "3-DICE-1-CLDICE"
        name:
        "3-DICE-1-CLDICE_03FMQ2460N"
        project:
        "vessels_combined_shifted_segmentation_prod"
      - type:
        "WandbVisBackend"
- WANDB_PROJECT_SUFFIX:
  "_prod"
- WANDB_RUN_NAME:
  "3-DICE-1-CLDICE"
- work_dir:
  "./work_dirs/config_vessels_4"

Summary metrics are your model's outputs. Learn more

▶
Summary metrics:{} 20 keys
- aux.acc_seg:
  83.26225280761719
- aux.loss_cl_dice:
  0.26382927149534224
- aux.loss_dice:
  0.4502255529165268
- avg:
  0.41450640806898775
- data_time:
  0.06267202311548693
- decode.acc_seg:
  86.7464599609375
- decode.loss_cl_dice:
  0.4728896826505661
- decode.loss_dice:
  0.9423988282680512
- dice:
  0.3634882939232843
- iou:
  0.22524178197809944
- iter:
  2,120
- length:
  0.48390561458242687
- line_distance:
  0.433231019696993
- loss:
  2.1293432831764223
- lr:
  0.00000015693659761456
- memory:
  18,862
- time:
  0.5047721698366362
- ▶
  val_17.png:{} 7 keys
- ▶
  val_26.png:{} 7 keys
- width:
  0.4410922887666798

This run consumed these artifacts as inputs. Learn more

job

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

This run produced these artifacts as outputs. Learn more

code

source-vessels_combined_shifted_segmentation_prod-mmsegmentation_tools_train.py:v40

wandb-history

run-cz8uhzm9-history:v0