You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Traceback (most recent call last):
File "tools/train.py", line 135, in
main()
File "tools/train.py", line 131, in main
runner.train()
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/runner.py", line 1777, in train
model = self.train_loop.run() # type: ignore
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/loops.py", line 98, in run
self.run_epoch()
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/loops.py", line 115, in run_epoch
self.run_iter(idx, data_batch)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/loops.py", line 131, in run_iter
outputs = self.runner.model.train_step(
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/model/base_model/base_model.py", line 114, in train_step
losses = self._run_forward(data, mode='loss') # type: ignore
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/model/base_model/base_model.py", line 361, in _run_forward
results = self(**data, mode=mode)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmdet3d/models/detectors/base.py", line 75, in forward
return self.loss(inputs, data_samples, **kwargs)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/oneformer3d.py", line 369, in loss
x = self.decoder(x, queries)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/query_decoder.py", line 340, in forward
return self.forward_iter_pred(x, queries)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/query_decoder.py", line 457, in forward_iter_pred
queries = self.cross_attn_layers[i](inst_feats, queries, attn_mask)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/query_decoder.py", line 52, in forward
output, _ = self.attn(queries[i], k, v, attn_mask=attn_mask)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/activation.py", line 1003, in forward
attn_output, attn_output_weights = F.multi_head_attention_forward(
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/functional.py", line 4967, in multi_head_attention_forward
tgt_len, bsz, embed_dim = query.shape
ValueError: not enough values to unpack (expected 3, got 2)
The text was updated successfully, but these errors were encountered:
Traceback (most recent call last):
File "tools/train.py", line 135, in
main()
File "tools/train.py", line 131, in main
runner.train()
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/runner.py", line 1777, in train
model = self.train_loop.run() # type: ignore
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/loops.py", line 98, in run
self.run_epoch()
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/loops.py", line 115, in run_epoch
self.run_iter(idx, data_batch)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/runner/loops.py", line 131, in run_iter
outputs = self.runner.model.train_step(
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/model/base_model/base_model.py", line 114, in train_step
losses = self._run_forward(data, mode='loss') # type: ignore
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmengine/model/base_model/base_model.py", line 361, in _run_forward
results = self(**data, mode=mode)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/mmdet3d/models/detectors/base.py", line 75, in forward
return self.loss(inputs, data_samples, **kwargs)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/oneformer3d.py", line 369, in loss
x = self.decoder(x, queries)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/query_decoder.py", line 340, in forward
return self.forward_iter_pred(x, queries)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/query_decoder.py", line 457, in forward_iter_pred
queries = self.cross_attn_layers[i](inst_feats, queries, attn_mask)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/oneformer3d-main/oneformer3d/query_decoder.py", line 52, in forward
output, _ = self.attn(queries[i], k, v, attn_mask=attn_mask)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/modules/activation.py", line 1003, in forward
attn_output, attn_output_weights = F.multi_head_attention_forward(
File "/root/miniconda3/envs/tr3d/lib/python3.8/site-packages/torch/nn/functional.py", line 4967, in multi_head_attention_forward
tgt_len, bsz, embed_dim = query.shape
ValueError: not enough values to unpack (expected 3, got 2)
The text was updated successfully, but these errors were encountered: