single-shot-robot

Josselin's group workspace

Group: Combined-4-DICE

1-20

of 20

Tags

Notes

Author

josselin

State

Finished

Start time

August 31st, 2023 9:48:47 PM

Runtime

26m 56s

Tracked hours

26m 48s

Run path

single-shot-robot/SKIL-Reachbot_vessels_combined_segmentation_prod/gev1y0a4

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 "Combined-4-DICE_1KTRQN4V1Q" 67be51cb5e8d8b81e19a20cb9540a7d9bd47f2d8

Command

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

Combined-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/vessels/combined"
- DATASET:
  "vessels_combined"
- 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:{} 4 keys
  - ▶
    logger:{} 3 keys
    - interval:
      5
    - log_metric_by_epoch:
      false
    - type:
      "LoggerHook"
  - ▶
    param_scheduler:{} 1 key
    - type:
      "ParamSchedulerHook"
  - ▶
    sampler_seed:{} 1 key
    - type:
      "DistSamplerSeedHook"
  - ▶
    timer:{} 1 key
    - type:
      "IterTimerHook"
- 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:
  5
- ▶
  LOSSES:{} 1 key
  - dice:
    4
- ▶
  losses_mappings:{} 5 keys
- LR:
  "aggressive"
- memory_available:
  47.15167236328125
- 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:
  "1KTRQN4V1Q"
- ▶
  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:
        "Combined-4-DICE"
        name:
        "Combined-4-DICE_1KTRQN4V1Q"
        project:
        "vessels_combined_segmentation_prod"
      - type:
        "WandbVisBackend"
- WANDB_PROJECT_SUFFIX:
  "_prod"
- WANDB_RUN_NAME:
  "Combined-4-DICE"
- work_dir:
  "./work_dirs/config_dice"

Summary metrics are your model's outputs. Learn more

▶
Summary metrics:{} 16 keys
- aux.acc_seg:
  89.03768157958984
- aux.loss_dice:
  0.5623992681503296
- avg:
  0.6194528380665286
- data_time:
  0.019353175163269044
- decode.acc_seg:
  94.27857971191406
- decode.loss_dice:
  0.8563850283622741
- dice:
  0.3997886537760502
- iou:
  0.2635201419535234
- iter:
  2,125
- length:
  0.6438355525176133
- line_distance:
  0.747676882640204
- loss:
  1.4187842845916747
- lr:
  0
- memory:
  15,097
- time:
  0.5393993854522705
- width:
  0.633388111273141

This run consumed these artifacts as inputs. Learn more

job

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

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

code

source-vessels_combined_segmentation_prod-mmsegmentation_tools_train.py:v70

wandb-history

run-gev1y0a4-history:v0