sony/nnabla-examples

train yolov2 error

Closed this issue · 1 comments

train_graph.py", line 63, in setup_impl
tcoord, mcoord, tconf, mconf, tcls, mcls = outputs
ValueError: not enough values to unpack (expected 6, got 1)

I guess you are using an older version of nnabla. Please upgrade nnabla and nnabla-ext-cuda as following;

pip uninstall -y nnabla nnabla-ext-cuda<CUDA_VERSION>
pip install nnabla nnabla-ext-cuda<CUDA_VERSION>