isl-org / ZoeDepth

Metric depth estimation from a single image
MIT License
2.36k stars 216 forks source link

IndexError: too many indices for tensor of dimension 4 #119

Closed dung2603 closed 4 months ago

dung2603 commented 5 months ago

Traceback (most recent call last): File "ZoeDepth-main/train_mono.py", line 169, in mp.spawn(main_worker, nprocs=ngpus_per_node, File "C:\Users\buiti\anaconda3\envs\gpu\lib\site-packages\torch\multiprocessing\spawn.py", line 239, in spawn return start_processes(fn, args, nprocs, join, daemon, start_method='spawn') File "C:\Users\buiti\anaconda3\envs\gpu\lib\site-packages\torch\multiprocessing\spawn.py", line 197, in start_processes while not context.join(): File "C:\Users\buiti\anaconda3\envs\gpu\lib\site-packages\torch\multiprocessing\spawn.py", line 160, in join raise ProcessRaisedException(msg, error_index, failed_process.pid) torch.multiprocessing.spawn.ProcessRaisedException:

-- Process 0 terminated with the following error: Traceback (most recent call last): File "C:\Users\buiti\anaconda3\envs\gpu\lib\site-packages\torch\multiprocessing\spawn.py", line 69, in _wrap fn(i, args) File "C:\Users\buiti\ZoeDepth-main\train_mono.py", line 105, in main_worker trainer.train() File "C:\Users\buiti\ZoeDepth-main\zoedepth\trainers\base_trainer.py", line 216, in train metrics, test_losses = self.validate() File "C:\Users\buiti\ZoeDepth-main\zoedepth\trainers\base_trainer.py", line 268, in validate metrics, losses = self.validate_on_batch(batch, val_step=i) File "C:\Users\buiti\ZoeDepth-main\zoedepth\trainers\zoedepth_trainer.py", line 166, in validate_on_batch l_depth = self.silog_loss( File "C:\Users\buiti\anaconda3\envs\gpu\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl return forward_call(args, **kwargs) File "C:\Users\buiti\ZoeDepth-main\zoedepth\trainers\loss.py", line 65, in forward input = input[mask] IndexError: too many indices for tensor of dimension 4