Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train erro #34

Closed
gostop112111 opened this issue Nov 19, 2018 · 1 comment
Closed

Train erro #34

gostop112111 opened this issue Nov 19, 2018 · 1 comment

Comments

@gostop112111
Copy link

gostop112111 commented Nov 19, 2018

The following error occurred while I was training coco.

Traceback (most recent call last):
  File "/project/yolov3/train.py", line 202, in <module>
    main(opt)
  File "/project/yolov3/train.py", line 132, in main
    loss = model(imgs.to(device), targets, requestPrecision=True)
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 123, in forward
    outputs = self.parallel_apply(replicas, inputs, kwargs)
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 133, in parallel_apply
    return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 77, in parallel_apply
    raise output
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 53, in _worker
    output = module(*input, **kwargs)
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/project/yolov3/models.py", line 238, in forward
    x, *losses = module[0](x, targets, requestPrecision)
  File "/data_b/VirEnv/project/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/project/yolov3/models.py", line 156, in forward
    requestPrecision)
  File "/project/yolov3/utils/utils.py", line 278, in build_targets
    tmp = pred_cls[b, a, gj, gi]
IndexError: index 8 is out of bounds for dimension 0 with size 8
@glenn-jocher
Copy link
Member

Only single-GPU operation is supported currently, see issue #21. Could this be your problem?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants