|
Traceback (most recent call last):
? File "/home/zhangyue/shiyan1/predict.py", line 67, in <module>
? ? r_image = frcnn.detect_image(image)
? File "/home/zhangyue/shiyan1/frcnn.py", line 143, in detect_image
? ? roi_cls_locs, roi_scores, rois, _ = self.net(images)
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
? ? result = self.forward(*input, **kwargs)
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 153, in forward
? ? return self.gather(outputs, self.output_device)
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 165, in gather
? ? return gather(outputs, output_device, dim=self.dim)
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 68, in gather
? ? res = gather_map(outputs)
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 63, in gather_map
? ? return type(out)(map(gather_map, zip(*outputs)))
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 55, in gather_map
? ? return Gather.apply(target_device, dim, *outputs)
? File "/home/zhangyue/anaconda3/envs/py36/lib/python3.6/site-packages/torch/nn/parallel/_functions.py", line 54, in forward
? ? assert all(map(lambda i: i.is_cuda, inputs))
AssertionError
|