是我太菜了 pso-elm 莫名其妙就报错了

Traceback (most recent call last): File "E:\Program Files\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\IPython\core\interactiveshell.py", line 3437, in run_code exec(code_obj, self.user_global_ns, self.user_ns) File "", line 1, in runfile('H:/pythonProject/pso_classed.py', wdir='H:/pythonProject') File "E:\Polyspace\PyCharm 2021.1\plugins\python\helpers\pydev_pydev_bundle\pydev_umd.py", line 197, in runfile pydev_imports.execfile(filename, global_vars, local_vars) # execute the script File "E:\Polyspace\PyCharm 2021.1\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "H:/pythonProject/pso_classed.py", line 96, in pso.main() File "H:/pythonProject/pso_classed.py", line 75, in main if self.fitness(self.g_best) > self.fitness(self.ng_best): File "H:/pythonProject/pso_classed.py", line 41, in fitness return e.aim_function(c) File "H:\pythonProject\elm.py", line 76, in aim_function model = Net(hidden_size=num, activation=args.activation) File "H:\pythonProject\elm.py", line 53, in init self.fc2 = nn.Linear(hidden_size, 10, bias=False) # ELM do not use bias in the output layer. File "E:\Program Files\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\linear.py", line 84, in init self.reset_parameters() File "E:\Program Files\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\linear.py", line 87, in reset_parameters init.kaiminguniform(self.weight, a=math.sqrt(5)) File "E:\Program Files\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\init.py", line 376, in kaiminguniform std = gain / math.sqrt(fan) ZeroDivisionError: float division by zero