Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RuntimeError: mat1 and mat2 shapes cannot be multiplied (0x1 and 64x128) #56

Open
linhcentrio opened this issue Sep 15, 2024 · 3 comments

Comments

@linhcentrio
Copy link

linhcentrio commented Sep 15, 2024

(D:\Gaussian\GaussianTalker\venv) D:\Gaussian\GaussianTalker>python train.py -s .\data\obama --model_path .\data\obama\models --configs arguments\64_dim_1_transformer.py
Optimizing .\data\obama\models
Output folder: .\data\obama\models [15/09 23:22:08]
feature_dim: 64 [15/09 23:22:08]
[INFO] load aud_features: torch.Size([7999, 29, 16]) [15/09 23:22:08]
Reading Training Transforms [15/09 23:22:09]
Reading Test Transforms [15/09 23:24:03]
Generating Video Transforms [15/09 23:24:18]
Loading Training Cameras [15/09 23:24:26]
Loading Test Cameras [15/09 23:24:26]
Loading Video Cameras [15/09 23:24:26]
Deformation Net Set aabb [0.75368124 0.8865968 0.49712884] [-0.7633047 -0.859515 -0.5761221] [15/09 23:24:26]
Voxel Plane: set aabb= Parameter containing:
tensor([[ 0.7537, 0.8866, 0.4971],
[-0.7633, -0.8595, -0.5761]]) [15/09 23:24:26]
Number of points at initialisation : 34650 [15/09 23:24:26]
['xyz', 'deformation', 'grid', 'f_dc', 'f_rest', 'opacity', 'scaling', 'rotation'] [15/09 23:24:26]
Training progress: 0%| | 0/7999 [00:00<?, ?it/s]data loading done [15/09 23:24:26]

[ITER 1] Saving Gaussians [15/09 23:35:54]
Training progress: 6%|█▏ | 500/7999 [12:10<10:14, 12.20it/s, Loss=0.2446528, psnr=20.31, point=34650]
[ITER 500] Saving Gaussians [15/09 23:36:36]
Training progress: 11%|██▌ | 900/7999 [12:41<08:50, 13.38it/s, Loss=0.2448057, psnr=20.14, point=2]Traceback (most recent call last):
File "D:\Gaussian\GaussianTalker\train.py", line 408, in
training(lp.extract(args), hp.extract(args), op.extract(args), pp.extract(args), args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from, args.expname, args.use_wandb)
File "D:\Gaussian\GaussianTalker\train.py", line 276, in training
scene_reconstruction(dataset, opt, hyper, pipe, testing_iterations, saving_iterations,
File "D:\Gaussian\GaussianTalker\train.py", line 153, in scene_reconstruction
output = render_from_batch(viewpoint_cams, gaussians, pipe, random_color, stage=stage, batch_size=batch_size, canonical_tri_plane_factor_list=opt.canonical_tri_plane_factor_list,iteration=iteration)
File "D:\Gaussian\GaussianTalker\gaussian_renderer_init_.py", line 101, in render_from_batch
means3D_final, scales_temp, rotations_temp, opacity_temp, shs_temp = pc._deformation(means3D, scales, rotations, opacity, shs, aud_features, eye_features, cam_features)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\scene\deformation.py", line 365, in forward
return self.forward_static(point, scales, rotations, opacity, shs)
File "D:\Gaussian\GaussianTalker\scene\deformation.py", line 375, in forward_static
points = self.deformation_net(points)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\scene\deformation.py", line 211, in forward
return self.forward_static(rays_pts_emb)
File "D:\Gaussian\GaussianTalker\scene\deformation.py", line 218, in forward_static
grid_feature, scale, rotation, opacity, sh = self.tri_plane(rays_pts_emb)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\scene\canonical_tri_plane.py", line 54, in forward
feature = self.feature_out(feature)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\container.py", line 215, in forward
input = module(input)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "D:\Gaussian\GaussianTalker\venv\lib\site-packages\torch\nn\modules\linear.py", line 114, in forward
return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (0x1 and 64x128)
Training progress: 11%|██▎ | 900/7999 [12:44<1:40:26, 1.18it/s, Loss=0.2448057, psnr=20.14, point=2]

@linhcentrio
Copy link
Author

packages in environment at D:\Gaussian\GaussianTalker\venv:

Name Version Build Channel

absl-py 2.1.0 pypi_0 pypi
addict 2.4.0 pypi_0 pypi
argparse 1.4.0 pypi_0 pypi
asttokens 2.4.1 pypi_0 pypi
astunparse 1.6.3 pypi_0 pypi
attrs 24.2.0 pypi_0 pypi
blinker 1.8.2 pypi_0 pypi
bzip2 1.0.8 hcfcfb64_5 conda-forge
ca-certificates 2024.8.30 h56e8100_0 conda-forge
cachetools 5.5.0 pypi_0 pypi
certifi 2022.12.7 pypi_0 pypi
cffi 1.17.1 pypi_0 pypi
charset-normalizer 2.1.1 pypi_0 pypi
click 8.1.7 pypi_0 pypi
colorama 0.4.6 pypi_0 pypi
comm 0.2.2 pypi_0 pypi
configargparse 1.7 pypi_0 pypi
contourpy 1.3.0 pypi_0 pypi
cuda-cccl 11.8.89 0 nvidia/label/cuda-11.8.0
cuda-command-line-tools 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-compiler 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-cudart 11.8.89 0 nvidia/label/cuda-11.8.0
cuda-cudart-dev 11.8.89 0 nvidia/label/cuda-11.8.0
cuda-cuobjdump 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-cupti 11.8.87 0 nvidia/label/cuda-11.8.0
cuda-cuxxfilt 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-documentation 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-libraries 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-libraries-dev 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-memcheck 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-nsight-compute 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-nvcc 11.8.89 0 nvidia/label/cuda-11.8.0
cuda-nvdisasm 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-nvml-dev 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-nvprof 11.8.87 0 nvidia/label/cuda-11.8.0
cuda-nvprune 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-nvrtc 11.8.89 0 nvidia/label/cuda-11.8.0
cuda-nvrtc-dev 11.8.89 0 nvidia/label/cuda-11.8.0
cuda-nvtx 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-nvvp 11.8.87 0 nvidia/label/cuda-11.8.0
cuda-profiler-api 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-sanitizer-api 11.8.86 0 nvidia/label/cuda-11.8.0
cuda-toolkit 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-tools 11.8.0 0 nvidia/label/cuda-11.8.0
cuda-version 12.5 hd4f0392_3 conda-forge
cuda-visual-tools 11.8.0 0 nvidia/label/cuda-11.8.0
cudnn 8.9.7.29 h3472fe1_3 conda-forge
cycler 0.12.1 pypi_0 pypi
dash 2.18.0 pypi_0 pypi
dash-core-components 2.0.0 pypi_0 pypi
dash-html-components 2.0.0 pypi_0 pypi
dash-table 5.0.0 pypi_0 pypi
dearpygui 1.11.1 pypi_0 pypi
decorator 5.1.1 pypi_0 pypi
diff-gaussian-rasterization 0.0.0 pypi_0 pypi
docker-pycreds 0.4.0 pypi_0 pypi
einops 0.8.0 pypi_0 pypi
exceptiongroup 1.2.2 pypi_0 pypi
executing 2.1.0 pypi_0 pypi
face-alignment 1.4.1 pypi_0 pypi
fastjsonschema 2.20.0 pypi_0 pypi
filelock 3.13.1 pypi_0 pypi
flask 3.0.3 pypi_0 pypi
flatbuffers 24.3.25 pypi_0 pypi
fonttools 4.53.1 pypi_0 pypi
fsspec 2024.2.0 pypi_0 pypi
fvcore 0.1.5.post20221221 pypi_0 pypi
gast 0.6.0 pypi_0 pypi
gitdb 4.0.11 pypi_0 pypi
gitpython 3.1.43 pypi_0 pypi
google-auth 2.34.0 pypi_0 pypi
google-auth-oauthlib 0.4.6 pypi_0 pypi
google-pasta 0.2.0 pypi_0 pypi
grpcio 1.66.1 pypi_0 pypi
h5py 3.11.0 pypi_0 pypi
idna 3.4 pypi_0 pypi
imageio 2.35.1 pypi_0 pypi
imageio-ffmpeg 0.5.1 pypi_0 pypi
importlib-metadata 8.4.0 pypi_0 pypi
iopath 0.1.10 pypi_0 pypi
ipython 8.27.0 pypi_0 pypi
ipywidgets 8.1.5 pypi_0 pypi
itsdangerous 2.2.0 pypi_0 pypi
jedi 0.19.1 pypi_0 pypi
jinja2 3.1.3 pypi_0 pypi
joblib 1.4.2 pypi_0 pypi
jsonschema 4.23.0 pypi_0 pypi
jsonschema-specifications 2023.12.1 pypi_0 pypi
jupyter-core 5.7.2 pypi_0 pypi
jupyterlab-widgets 3.0.13 pypi_0 pypi
keras 2.8.0 pypi_0 pypi
keras-preprocessing 1.1.2 pypi_0 pypi
kiwisolver 1.4.7 pypi_0 pypi
lazy-loader 0.4 pypi_0 pypi
libclang 18.1.1 pypi_0 pypi
libcublas 11.11.3.6 0 nvidia/label/cuda-11.8.0
libcublas-dev 11.11.3.6 0 nvidia/label/cuda-11.8.0
libcufft 10.9.0.58 0 nvidia/label/cuda-11.8.0
libcufft-dev 10.9.0.58 0 nvidia/label/cuda-11.8.0
libcurand 10.3.0.86 0 nvidia/label/cuda-11.8.0
libcurand-dev 10.3.0.86 0 nvidia/label/cuda-11.8.0
libcusolver 11.4.1.48 0 nvidia/label/cuda-11.8.0
libcusolver-dev 11.4.1.48 0 nvidia/label/cuda-11.8.0
libcusparse 11.7.5.86 0 nvidia/label/cuda-11.8.0
libcusparse-dev 11.7.5.86 0 nvidia/label/cuda-11.8.0
libffi 3.4.2 h8ffe710_5 conda-forge
libnpp 11.8.0.86 0 nvidia/label/cuda-11.8.0
libnpp-dev 11.8.0.86 0 nvidia/label/cuda-11.8.0
libnvjpeg 11.9.0.86 0 nvidia/label/cuda-11.8.0
libnvjpeg-dev 11.9.0.86 0 nvidia/label/cuda-11.8.0
libsqlite 3.46.0 h2466b09_0 conda-forge
libzlib 1.3.1 h2466b09_1 conda-forge
libzlib-wapi 1.2.13 h2466b09_6 conda-forge
llvmlite 0.43.0 pypi_0 pypi
lpips 0.1.4 pypi_0 pypi
markdown 3.7 pypi_0 pypi
markupsafe 2.1.5 pypi_0 pypi
matplotlib 3.9.2 pypi_0 pypi
matplotlib-inline 0.1.7 pypi_0 pypi
mmcv 1.6.0 pypi_0 pypi
mpmath 1.3.0 pypi_0 pypi
nbformat 5.10.4 pypi_0 pypi
nest-asyncio 1.6.0 pypi_0 pypi
networkx 3.2.1 pypi_0 pypi
ninja 1.12.1 hc790b64_0 conda-forge
nsight-compute 2022.3.0.22 0 nvidia/label/cuda-11.8.0
numba 0.60.0 pypi_0 pypi
numpy 1.26.3 pypi_0 pypi
oauthlib 3.2.2 pypi_0 pypi
open3d 0.18.0 pypi_0 pypi
opencv-python 4.10.0.84 pypi_0 pypi
openssl 3.3.2 h2466b09_0 conda-forge
opt-einsum 3.3.0 pypi_0 pypi
packaging 24.1 pypi_0 pypi
pandas 2.2.2 pypi_0 pypi
parso 0.8.4 pypi_0 pypi
pillow 10.2.0 pypi_0 pypi
pip 24.0 pyhd8ed1ab_0 conda-forge
platformdirs 4.2.2 pypi_0 pypi
plotly 5.24.0 pypi_0 pypi
plyfile 0.8.1 pypi_0 pypi
portalocker 2.10.1 pypi_0 pypi
prompt-toolkit 3.0.47 pypi_0 pypi
protobuf 3.20.1 pypi_0 pypi
psutil 6.0.0 pypi_0 pypi
pure-eval 0.2.3 pypi_0 pypi
pyasn1 0.6.0 pypi_0 pypi
pyasn1-modules 0.4.0 pypi_0 pypi
pyaudio 0.2.14 pypi_0 pypi
pycparser 2.22 pypi_0 pypi
pygments 2.18.0 pypi_0 pypi
pymcubes 0.1.6 pypi_0 pypi
pyparsing 3.1.4 pypi_0 pypi
python 3.10.14 h4de0772_0_cpython conda-forge
python-dateutil 2.9.0.post0 pypi_0 pypi
python-speech-features 0.6 pypi_0 pypi
pytorch-msssim 1.0.0 pypi_0 pypi
pytorch3d 0.7.7+pt2.1.2cu118 pypi_0 pypi
pytz 2024.1 pypi_0 pypi
pywin32 306 pypi_0 pypi
pyyaml 6.0.2 pypi_0 pypi
referencing 0.35.1 pypi_0 pypi
regex 2024.7.24 pypi_0 pypi
requests 2.28.1 pypi_0 pypi
requests-oauthlib 2.0.0 pypi_0 pypi
resampy 0.4.3 pypi_0 pypi
retrying 1.3.4 pypi_0 pypi
rpds-py 0.20.0 pypi_0 pypi
rsa 4.9 pypi_0 pypi
scikit-image 0.24.0 pypi_0 pypi
scikit-learn 1.5.1 pypi_0 pypi
scipy 1.14.1 pypi_0 pypi
sentry-sdk 2.13.0 pypi_0 pypi
setproctitle 1.3.3 pypi_0 pypi
setuptools 74.1.1 pypi_0 pypi
simple-knn 0.0.0 pypi_0 pypi
six 1.16.0 pypi_0 pypi
smmap 5.0.1 pypi_0 pypi
soundfile 0.12.1 pypi_0 pypi
stack-data 0.6.3 pypi_0 pypi
sympy 1.12 pypi_0 pypi
tabulate 0.9.0 pypi_0 pypi
tenacity 9.0.0 pypi_0 pypi
tensorboard 2.8.0 pypi_0 pypi
tensorboard-data-server 0.6.1 pypi_0 pypi
tensorboard-plugin-wit 1.8.1 pypi_0 pypi
tensorboardx 2.6.2.2 pypi_0 pypi
tensorflow-gpu 2.8.0 pypi_0 pypi
tensorflow-io-gcs-filesystem 0.31.0 pypi_0 pypi
termcolor 2.4.0 pypi_0 pypi
tf-estimator-nightly 2.8.0.dev2021122109 pypi_0 pypi
threadpoolctl 3.5.0 pypi_0 pypi
tifffile 2024.8.30 pypi_0 pypi
tk 8.6.13 h5226925_1 conda-forge
tomli 2.0.1 pypi_0 pypi
torch 2.1.2+cu118 pypi_0 pypi
torch-ema 0.3 pypi_0 pypi
torchaudio 2.1.2+cu118 pypi_0 pypi
torchvision 0.16.2+cu118 pypi_0 pypi
tqdm 4.66.5 pypi_0 pypi
traitlets 5.14.3 pypi_0 pypi
trimesh 4.4.9 pypi_0 pypi
typing-extensions 4.9.0 pypi_0 pypi
tzdata 2024.1 pypi_0 pypi
ucrt 10.0.22621.0 h57928b3_0 conda-forge
urllib3 1.26.13 pypi_0 pypi
vc 14.3 h8a93ad2_20 conda-forge
vc14_runtime 14.40.33810 ha82c5b3_20 conda-forge
vs2015_runtime 14.40.33810 h3bf8584_20 conda-forge
vs2019_win-64 19.29.30154 h96f319f_5 anaconda
vswhere 2.8.4 haa95532_0 anaconda
wandb 0.17.8 pypi_0 pypi
wcwidth 0.2.13 pypi_0 pypi
werkzeug 3.0.4 pypi_0 pypi
wget 3.2 pypi_0 pypi
wheel 0.44.0 pypi_0 pypi
widgetsnbextension 4.0.13 pypi_0 pypi
wrapt 1.16.0 pypi_0 pypi
xz 5.2.6 h8d14728_0 conda-forge
yacs 0.1.8 pypi_0 pypi
yapf 0.40.2 pypi_0 pypi
zipp 3.20.1 pypi_0 pypi

@nguyenchiminhhieu
Copy link

hello. I have some issues I would like to ask you about setting up an environment for this project. How can I contact you?

@czw612
Copy link

czw612 commented Oct 26, 2024

I added a linear layer to align the two matrices.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants