Skip to content

quangnh-2761/sun-polyp

 
 

Repository files navigation

Config

Config everything in mcode/config.py
What to config?

  • Model:
    • Follow mmseg config
    • pretrained: path to ImageNet pretrained MiT backbone
    • Please change pretrained in backbone to pretrained=pretrained
    • Config model head to head of your choice
  • Wandb:
    • use_wandb: True, False if debug
    • wandb_key: Please use your wandb authorize key
    • wandb_name: Name of your experiments, please make it as specific as possible
    • wandb_group: We need 5 runs/experiments, grouping makes it easier to see on wandb UI
  • Dataset:
    • train_images: path to image in Train Dataset
    • train_masks: path to mask in Train Dataset
    • test_folder: path to Test Dataset
    • test_images and test_masks: leave it
    • num_workers: torch workers
    • save_path: path to save checkpoints and logs
    • bs: this should be 16 if possible
    • grad_accumulate_rate: num iters to backward, if bs=16, this should be 1

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.9%
  • Shell 0.1%