Skip to content

Questions About running test but got IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1) #37

Description

@GenPilot-Any

ALL the models are set following instructions in readme

I occur this error

Traceback (most recent call last):
  File "launch.py", line 239, in <module>
    main(args, extras)
  File "launch.py", line 194, in main
    trainer.test(system, datamodule=dm, ckpt_path=cfg.resume)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 742, in test
    return call._call_and_handle_interrupt(
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/trainer/call.py", line 43, in _call_and_handle_interrupt
    return trainer_fn(*args, **kwargs)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 785, in _test_impl
    results = self._run(model, ckpt_path=ckpt_path)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 980, in _run
    results = self._run_stage()
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1016, in _run_stage
    return self._evaluation_loop.run()
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/loops/utilities.py", line 181, in _decorator
    return loop_run(self, *args, **kwargs)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/loops/evaluation_loop.py", line 115, in run
    self._evaluation_step(batch, batch_idx, dataloader_idx)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/loops/evaluation_loop.py", line 376, in _evaluation_step
    output = call._call_strategy_hook(trainer, hook_name, *step_kwargs.values())
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/trainer/call.py", line 294, in _call_strategy_hook
    output = fn(*args, **kwargs)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/pytorch_lightning/strategies/strategy.py", line 403, in test_step
    return self.model.test_step(*args, **kwargs)
  File "/root/HumanGaussian/threestudio/systems/GaussianDreamer.py", line 460, in test_step
    out = self(batch,testbackground_tensor)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/root/HumanGaussian/threestudio/systems/GaussianDreamer.py", line 248, in forward
    render_pkg = render(viewpoint_cam, self.gaussian, self.pipe, renderbackground)
  File "/root/HumanGaussian/gaussiansplatting/gaussian_renderer/__init__.py", line 66, in render
    rotations = pc.get_rotation
  File "/root/HumanGaussian/gaussiansplatting/scene/gaussian_model.py", line 102, in get_rotation
    return self.rotation_activation(self._rotation)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/torch/nn/functional.py", line 4660, in normalize
    denom = input.norm(p, dim, keepdim=True).clamp_min(eps).expand_as(input)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/torch/_tensor.py", line 647, in norm
    return torch.norm(self, p, dim, keepdim, dtype=dtype)
  File "/root/miniconda3/envs/humangussian/lib/python3.8/site-packages/torch/functional.py", line 1517, in norm
    return torch.linalg.vector_norm(input, _p, _dim, keepdim, dtype=dtype)
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

when running python launch.py --config configs/test.yaml --test --gpu 0 system.prompt_processor.prompt="A boy with a beanie wearing a hoodie and joggers"

Anyone help? It is very urgent

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions