chainer-pspnet
chainer-pspnet copied to clipboard
Training Fails with IndexError when using train.py
Hi, I got following error when I tried to train PSP net with your train.py
How can I train my PSPNet model?
root@9bdbca328f5f:/yendo/oss/chainer-pspnet# python3 train.py --gpu --result_dir result configs/cityscapes/pspnet.yml
==========================================
Chainer version: 3.4.0
CuPy version: 2.4.0
Traceback (most recent call last):
File "train.py", line 483, in <module>
trainer = get_trainer(args)
File "train.py", line 339, in get_trainer
chainer.cuda.available, chainer.cuda.cudnn_enabled, ))
IndexError: tuple index out of range