Traceback (most recent call last):


 "<stdin>", line 1, in <module>


 "/Users/saxen/.julia/conda/3/lib/python3.8/site-packages/arviz/plots/densityplot.py", line 252, in plot_density

ax = plot(**plot_density_kwargs)


 "/Users/saxen/.julia/conda/3/lib/python3.8/site-packages/arviz/plots/backends/bokeh/densityplot.py", line 74, in plot_density

for label, ax_ in zip(all_labels, (item for item in ax.flatten() if item is not None))
AttributeError: 'Figure' object has no attribute 'flatten'

Traceback (most recent call last):


 "demo.py", line 56, in <module>

demo(opt)


 "demo.py", line 49, in demo

ret = detector.run(image_name)


 "/workspace/midhilesh/CenterNet/src/lib/detectors/base_detector.py", line 116, in run

output, dets, forward_time = self.process(images, return_time=True)


 "/workspace/midhilesh/CenterNet/src/lib/detectors/ctdet.py", line 30, in process

output = self.model(images)[-1]


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/workspace/midhilesh/CenterNet/src/lib/models/networks/pose_dla_dcn.py", line 471, in forward

x = self.base(x)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/workspace/midhilesh/CenterNet/src/lib/models/networks/pose_dla_dcn.py", line 288, in forward

x = self.base_layer(x)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward

input = module(input)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__

result = self.forward(*input, **kwargs)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward

self.padding, self.dilation, self.groups)
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
