-
Notifications
You must be signed in to change notification settings - Fork 6
Description
Start training...
Start Training Epoch 0 : 2023-03-25 16:48:34
0%| | 0/2206 [00:01<?, ?it/s]
Traceback (most recent call last):
File "main.py", line 158, in
graphs_pattern, graphs_others, nums_nodes)
File "/home/cv/miniconda3/envs/Zmd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/cv/ZMD/Pref-FEND-main/model/PrefFEND.py", line 100, in forward
idxs, dataset, tokens_features, map_for_fact_detector=map_entity, map_for_pattern_detector=map_pattern, fcs=self.fcs)
File "/home/cv/ZMD/Pref-FEND-main/model/PrefFEND.py", line 62, in forward_PreferencedDetector
idxs, dataset, tokens_features, map_for_pattern_detector)
File "/home/cv/miniconda3/envs/Zmd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/cv/ZMD/Pref-FEND-main/model/PatternBasedModels.py", line 131, in forward
out = self.forward_BERT(idxs, dataset, nodes_tokens, maps=tokened_maps)
File "/home/cv/ZMD/Pref-FEND-main/model/PatternBasedModels.py", line 163, in forward_BERT
out = self.fc(output)
File "/home/cv/miniconda3/envs/Zmd/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/cv/miniconda3/envs/Zmd/lib/python3.6/site-packages/torch/nn/modules/linear.py", line 93, in forward
return F.linear(input, self.weight, self.bias)
File "/home/cv/miniconda3/envs/Zmd/lib/python3.6/site-packages/torch/nn/functional.py", line 1690, in linear
ret = torch.addmm(bias, input, weight.t())
RuntimeError: mat1 dim 1 must match mat2 dim 0
似乎是bert维度的问题,请问大佬要怎么设置呢