关于NeuEvo的问题

[GreyRaven1999]

您好，我在尝试复现examples/Perception_and_Learning/NeuEvo/train_search.py时遇到了如下报错：
Traceback (most recent call last):
File "/home/dell/data2/zh/SNN/Evolution/NeuEvo/train_search.py", line 381, in
main()
File "/home/dell/data2/zh/SNN/Evolution/NeuEvo/train_search.py", line 256, in main
genotype = model.genotype()
File "/home/dell/anaconda3/envs/Neural-Network-zh/lib/python3.10/site-packages/braincog/model_zoo/NeuEvo/model_search.py", line 307, in genotype
gene_normal = parse(calc_weight(self.alphas_normal).data.cpu().numpy(),
File "/home/dell/anaconda3/envs/Neural-Network-zh/lib/python3.10/site-packages/braincog/model_zoo/NeuEvo/model_search.py", line 16, in calc_weight
tmp0 = torch.split(x[0], edge_num, dim=0)
File "/home/dell/anaconda3/envs/Neural-Network-zh/lib/python3.10/site-packages/torch/functional.py", line 196, in split
return tensor.split(split_size_or_sections, dim)
File "/home/dell/anaconda3/envs/Neural-Network-zh/lib/python3.10/site-packages/torch/_tensor.py", line 917, in split
return torch._VF.split_with_sizes(self, split_size, dim)
RuntimeError: split_with_sizes expects split_sizes to sum exactly to 14 (input tensor's size at dimension 0), but got split_sizes=[2, 3, 4, 1, 2]
ERROR conda.cli.main_run:execute(124): conda run python /home/dell/data2/zh/SNN/Evolution/NeuEvo/train_search.py failed. (See above for error)
请问是from braincog.model_zoo.NeuEvo.architect import Architect中所依赖的Architect.py和train_search.py代码不匹配导致的吗？