Sizes of tensors must match except in dimension 1. Expected size 832 but got size 416 for tensor
number 1 in the list.
Error occurs, No graph saved
Traceback (most recent call last):
File "/home/ubuntu/桌面/singleshot6Dpose-master/train.py", line 442, in <module>
niter = train(epoch)
File "/home/ubuntu/桌面/singleshot6Dpose-master/train.py", line 78, in train
tb_writer.add_graph(model, init_image)
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/utils/tensorboard
/writer.py", line 736, in add_graph
self._get_file_writer().add_graph(graph(model, input_to_model, verbose, use_strict_trace))
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/utils/tensorboard
/_pytorch_graph.py", line 295, in graph
raise e
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/utils/tensorboard
/_pytorch_graph.py", line 289, in graph
trace = torch.jit.trace(model, args, strict=use_strict_trace)
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/jit/_trace.py",
line 750, in trace
_module_class,
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/jit/_trace.py",
line 965, in trace_module
argument_names,
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/nn/modules/mo
dule.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File
"/home/ubuntu/anaconda3/envs/fcn-torch/lib/python3.6/site-packages/torch/nn/modules/mo
dule.py", line 1090, in _slow_forward
result = self.forward(*input, **kwargs)
File "/home/ubuntu/桌面/singleshot6Dpose-master/darknet.py", line 1281, in forward
x = torch.cat((x1, x2), 1)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 832 but got size
416 for tensor number 1 in the list.
- 1
- 2
前往页