Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Training on ADE20K

See original GitHub issue

Hello! I try to train default models from train.py on ADE20K, but I see such error:

Input arguments:
weights_decoder  
batch_size_per_gpu 2
weights_encoder  
workers          16
lr_pow           0.9
seed             304
epoch_iters      5000
weight_decay     0.0001
root_dataset     ./data/
list_train       ./data/train.odgt
optim            SGD
num_epoch        20
ckpt             ./ckpt
list_val         ./data/validation.odgt
id               baseline
imgMaxSize       1000
deep_sup_scale   0.4
fix_bn           False
lr_encoder       0.02
gpus             0
beta1            0.9
random_flip      True
num_class        150
start_epoch      1
imgSize          [300, 375, 450, 525, 600]
arch_decoder     ppm_deepsup
disp_iter        20
padding_constant 8
segm_downsampling_rate 8
lr_decoder       0.02
fc_dim           2048
arch_encoder     resnet50dilated
Model ID: baseline-resnet50dilated-ppm_deepsup-ngpus1-batchSize2-imgMaxSize1000-paddingConst8-segmDownsampleRate8-LR_encoder0.02-LR_decoder0.02-epoch20
# samples: 20210
1 Epoch = 5000 iters
Traceback (most recent call last):
  File "train.py", line 325, in <module>
    main(args)
  File "train.py", line 200, in main
    train(segmentation_module, iterator_train, optimizers, history, epoch, args)
  File "train.py", line 37, in train
    loss, acc = segmentation_module(batch_data)
  File "/home/ponomareva/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ponomareva/semantic-segmentation-pytorch/models/models.py", line 34, in forward
    (pred, pred_deepsup) = self.decoder(self.encoder(feed_dict['img_data'], return_feature_maps=True))
TypeError: list indices must be integers or slices, not str

Issue Analytics

State:
Created 4 years ago
Reactions:2
Comments:5 (1 by maintainers)

Top GitHub Comments

3reactions

ShihuaiXucommented, Sep 12, 2019

@hangzhaomit thanks your help and now this project supports single gpu?

0reactions

adityashrm21commented, May 20, 2020

@loveis98 You might want to try this: https://github.com/CSAILVision/semantic-segmentation-pytorch/issues/203#issuecomment-562524601

Top Results From Across the Web

5. Train PSPNet on ADE20K Dataset

This is a tutorial of training PSPNet on ADE20K dataset using Gluon Vison. The readers should have basic knowledge of deep learning and...

ADE20K dataset

Training set. 25.574 images. All images are fully annotated with objects and, many of the images have parts too.

Semantic Segmentation on ADE20K - Papers With Code

Rank Model Validation mIoU Year Tags 1 InternImage‑DCNv3‑H (M3I Pre‑training) 62.9 2022 DCNDeformable Co... 2 M3I Pre‑training (InternImage‑H) 62.9 2022 3 BEiT‑3 62.8 2022 ViT‑Adapter

Review — ADE20K: Semantic Understanding of Scenes ...

ADE20K dataset is constructed as the benchmarks for scene parsing and ... After annotation, there are 20,210 images in the training set, ...

Setting a Baseline for Image Segmentation Speedups

By the end, we demonstrate a DeepLabv3+ baseline on ADE20k with +1.4 ... and a 1.8x faster training time than previously published baselines....