Inference Error #2
Description
Hello, executing python layout.py --image_path bedroom.jpg results in following error
Testing DataLoader 0: 0%| | 0/1 [00:00<?, ?it/s]Traceback (most recent call last):
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/layout.py", line 138, in <module>
main(args.image_path)
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/layout.py", line 29, in main
main_start()
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/spvloc/spvloc_train_test.py", line 83, in main_start
trainer.test(model)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/trainer/trainer.py", line 748, in test
return call._call_and_handle_interrupt(
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/trainer/call.py", line 47, in _call_and_handle_interrupt
return trainer_fn(*args, **kwargs)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/trainer/trainer.py", line 788, in _test_impl
results = self._run(model, ckpt_path=ckpt_path)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/trainer/trainer.py", line 981, in _run
results = self._run_stage()
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/trainer/trainer.py", line 1018, in _run_stage
return self._evaluation_loop.run()
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/loops/utilities.py", line 178, in _decorator
return loop_run(self, *args, **kwargs)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/loops/evaluation_loop.py", line 135, in run
self._evaluation_step(batch, batch_idx, dataloader_idx, dataloader_iter)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/loops/evaluation_loop.py", line 396, in _evaluation_step
output = call._call_strategy_hook(trainer, hook_name, *step_args)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/trainer/call.py", line 319, in _call_strategy_hook
output = fn(*args, **kwargs)
File "/home/ubuntu/miniconda3/envs/pixht/lib/python3.9/site-packages/pytorch_lightning/strategies/strategy.py", line 424, in test_step
return self.lightning_module.test_step(*args, **kwargs)
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/spvloc/spvloc/model/spvloc.py", line 298, in test_step
output, output_plot = self.test_step_full_epoch(
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/spvloc/spvloc/model/spvloc.py", line 545, in test_step_full_epoch
max_score, xcors, bbs_est, vp_masks_est, (pano_encode_time, correlation_time) = self.split_chunks(
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/spvloc/spvloc/model/spvloc.py", line 486, in split_chunks
) = self.bounding_box_decoder.forward_regress_boxes(pano_chunk, features_chunk)
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/spvloc/spvloc/model/modules.py", line 573, in forward_regress_boxes
transformed_anchors = self.regressBoxes.forward(anchors, regression)
File "/home/ubuntu/training/Layot/2D-Room-Layout-Estimation/spvloc/spvloc/model/modules.py", line 386, in forward
pred_ctr_x = ctr_x + dx * widths
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
Testing DataLoader 0: 0%| | 0/1 [00:04<?, ?it/s]