<jupyter_start><jupyter_text>Train Model<jupyter_code>!python3 train.py --gpus 0 --cfg config/ade20k-hrnetv2.yaml \
DATASET.list_train "./data/training.odgt" DATASET.list_val "./data/validation.odgt" \
MODEL.object_index 0 \
TRAIN.batch_size_per_gpu 4 \
VAL.batch_size 4 \
TRAIN.num_epoch 60
# TRAIN.start_epoch 0 \
# TRAIN.load_unet False<jupyter_output>[2020-12-07 05:41:09,749 INFO train.py line 460 1434] Loaded configuration file config/ade20k-hrnetv2.yaml
[2020-12-07 05:41:09,750 INFO train.py line 461 1434] Running with config:
DATASET:
  imgMaxSize: 1000
  imgSizes: (300, 375, 450, 525, 600)
  list_train: ./data/training.odgt
  list_val: ./data/validation.odgt
  num_class: 150
  padding_constant: 32
  random_flip: True
  root_dataset: ./data/
  segm_downsampling_rate: 4
DIR: ckpt/ade20k-hrnetv2-c1
MODEL:
  arch_decoder: c1
  arch_encoder: hrnetv2
  fc_dim: 720
  object_index: 0
  weights_decoder: 
  weights_encoder: 
  weights_unet: 
TEST:
  batch_size: 1
  checkpoint: epoch_30.pth
  result: ./
TRAIN:
  batch_size_per_gpu: 4
  beta1: 0.9
  deep_sup_scale: 0.4
  disp_iter: 20
  epoch_iters: 5000
  fix_bn: False
  load_unet: True
  lr_decoder: 0.02
  lr_encoder: 0.02
  lr_pow: 0.9
  lr_unet: 0.02
  num_epoch: 60
  optim: SGD
  seed: 304
  start_epoch: 0
  weight_decay: 0.0001
  workers: 16
VAL:
  batch_size: 4
  checkpoint: epoch_3[...]
{'fraction_non_alphanumeric': 0.10588235294117647, 'fraction_numerical': 0.10147058823529412, 'mean_word_length': 3.8607142857142858, 'pattern_counts': {'":': 0, '<': 4, '<?xml version=': 0, '>': 4, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '1', 'dir': 'jupyter-structured-clean-dedup', 'id': '368744', 'license': 'BSD-3-Clause', 'n_tokens_mistral': 686, 'n_tokens_neox': 575, 'n_words': 126, 'path': 'SSP_plus_weights_Wall.ipynb', 'repo_name': 'jaytoone/Indoor_Segmentation', 'size': '1360'}