About eval_configs about mace HOT 6 CLOSED

Traceback (most recent call last):
  File "/u/mncui/software/mace_v2/scripts/eval_configs.py", line 113, in <module>
    main()
  File "/u/mncui/software/mace_v2/scripts/eval_configs.py", line 78, in main
    output = model(batch, training=False)
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/mace/modules/models.py", line 226, in forward
    edge_index=data.edge_index,
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/mace/modules/blocks.py", line 517, in forward
    node_feats[sender], edge_attrs, tp_weights
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/e3nn/o3/_tensor_product/_tensor_product.py", line 529, in forward
    return self._compiled_main_left_right(x, y, real_weight)
  File "/u/mncui/software/anaconda3/envs/mace_env2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript, serialized code (most recent call last):
  File "code/__torch__/torch/fx/graph_module/___torch_mangle_8.py", line 109, in forward
    einsum_18 = torch.einsum("dba,dbc->dbac", [reshape_5, reshape_28])
    mul_2 = torch.mul(reshape_7, 2.2360679774997898)
    tensordot_1 = torch.tensordot(mul_2, _w3j_1_1_2, [2], [1])
                  ~~~~~~~~~~~~~~~ <--- HERE
    einsum_19 = torch.einsum("edab,ecad->ecb", [tensordot_1, einsum_18])
    reshape_29 = torch.reshape(einsum_19, [getitem_4, 640])

Traceback of TorchScript, original code (most recent call last):
    einsum_18 = torch.functional.einsum('dba,dbc->dbac', reshape_5, reshape_28);  reshape_28 = None
    mul_2 = reshape_7 * 2.23606797749979;  reshape_7 = None
    tensordot_1 = torch.functional.tensordot(mul_2, _w3j_1_1_2, [2], [1]);  mul_2 = _w3j_1_1_2 = None
                  ~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
    einsum_19 = torch.functional.einsum('edab,ecad->ecb', tensordot_1, einsum_18);  tensordot_1 = einsum_18 = None
    reshape_29 = einsum_19.reshape(getitem_4, 640);  einsum_19 = None
RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)