how to run distill_train.py?

[Liszt35]

thanks for the great code!
Please could you provide clear instructions for distill_train.py.I meet some problem when I run distill_train.py with custom dataset.

I use this instruction 'python distill_train.py -m D:\Nerf\LightGaussian-main\output\2ac6a73b-4 -s D:\Nerf\LightGaussian-main\data\ggbond --teacher_model D:\Nerf\LightGaussian-main\output\2ac6a73b-4\chkpnt30000.pt'

here is my teminal:
Optimizing D:\Nerf\LightGaussian-main\output\2ac6a73b-4
Output folder: D:\Nerf\LightGaussian-main\output\2ac6a73b-4 [24/07 15:25:07]
Tensorboard not available: not logging progress [24/07 15:25:07]
D:\Nerf\LightGaussian-main\data\ggbond [24/07 15:25:07]
Reading camera 93/93 [24/07 15:25:07]
Loading Training Cameras [24/07 15:25:07]
[ INFO ] Encountered quite large input images (>1.6K pixels width), rescaling to 1.6K.
If this is not desired, please explicitly specify '--resolution/-r' as 1 [24/07 15:25:07]
Loading Test Cameras [24/07 15:25:16]
Number of points at initialisation : 6272 [24/07 15:25:16]
Training progress: 0%| | 0/30000 [00:00<?, ?it/s]Traceback (most recent call last):
File "D:\Nerf\LightGaussian-main\distill_train.py", line 219, in
training(args, lp.extract(args), op.extract(args), pp.extract(args), args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from, args.new_max_sh)
File "D:\Nerf\LightGaussian-main\distill_train.py", line 141, in training
teacher_image = render(viewpoint_cam, teacher_gaussians, pipe, background)["render"].detach()
File "D:\Nerf\LightGaussian-main\gaussian_renderer_init_.py", line 83, in render
rotations = pc.get_rotation
File "D:\Nerf\LightGaussian-main\scene\gaussian_model.py", line 104, in get_rotation
return self.rotation_activation(self.rotation)
File "D:\Anaconda\Anaconda3\envs\light_gs\lib\site-packages\torch\nn\functional.py", line 4620, in normalize
teacher_image = render(viewpoint_cam, teacher_gaussians, pipe, background)["render"].detach()
File "D:\Nerf\LightGaussian-main\gaussian_renderer_init.py", line 83, in render
rotations = pc.get_rotation
File "D:\Nerf\LightGaussian-main\scene\gaussian_model.py", line 104, in get_rotation
return self.rotation_activation(self._rotation)
File "D:\Anaconda\Anaconda3\envs\light_gs\lib\site-packages\torch\nn\functional.py", line 4620, in normalize
denom = input.norm(p, dim, keepdim=True).clamp_min(eps).expand_as(input)
File "D:\Anaconda\Anaconda3\envs\light_gs\lib\site-packages\torch_tensor.py", line 531, in norm
return torch.norm(self, p, dim, keepdim, dtype=dtype)
File "D:\Anaconda\Anaconda3\envs\light_gs\lib\site-packages\torch\functional.py", line 1495, in norm
return _VF.norm(input, p, _dim, keepdim=keepdim) # type: ignore[attr-defined]
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
Training progress: 0%|

I got this result after run prune_finetune.py.

[seulqxq]

Hi, have you solved this problem?