Batch_normalize
Created by: pyqpyqpyqpyq789
File "train.py", line 363, in main
res = train(opt=options,
File "train.py", line 175, in train
model_output = model(x)
File "/home/user/anaconda3/envs/pyq38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/devdata/pyq/DASTM-main/protonet.py", line 196, in forward
x = self.model(x)
File "/home/user/anaconda3/envs/pyq38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/devdata/pyq/DASTM-main/mmskl/st_gcn_aaai18.py", line 124, in forward
x = self.data_bn(x)
File "/home/user/anaconda3/envs/pyq38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/user/anaconda3/envs/pyq38/lib/python3.8/site-packages/torch/nn/modules/batchnorm.py", line 131, in forward
return F.batch_norm(
File "/home/user/anaconda3/envs/pyq38/lib/python3.8/site-packages/torch/nn/functional.py", line 2056, in batch_norm
return torch.batch_norm(
RuntimeError: running_mean should contain 54 elements not 75