cyrusyc

/

mace-universal

like 6

• "__torch__.e3nn.math._normalize_activation.___torch_mangle_18.normalize2mom",
• "__torch__.e3nn.math._normalize_activation.___torch_mangle_57.normalize2mom",
• "__torch__.e3nn.nn._fc._Layer",
• "collections.OrderedDict",
• "__torch__.mace.modules.blocks.___torch_mangle_54.EquivariantProductBasisBlock",
• "__torch__.torch.fx.graph_module.___torch_mangle_31.GraphModule",
• "__torch__.e3nn.nn._fc.___torch_mangle_17._Layer",
• "__torch__.torch.fx.graph_module.___torch_mangle_7.tp_forward",
• "__torch__.mace.modules.radial.BesselBasis",
• "__torch__.e3nn.o3._linear.___torch_mangle_0.Linear",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.fx.graph_module.___torch_mangle_12.tp_forward",
• "__torch__.torch.fx.graph_module.tp_forward",
• "__torch__.torch.nn.modules.container.ModuleList",
• "torch.DoubleStorage",
• "__torch__.mace.modules.blocks.___torch_mangle_24.RealAgnosticResidualInteractionBlock",
• "__torch__.torch.fx.graph_module.___torch_mangle_0.GraphModule",
• "__torch__.e3nn.o3._linear.___torch_mangle_62.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_49.ModuleList",
• "__torch__.torch.fx.graph_module.___torch_mangle_13.GraphModule",
• "__torch__.mace.modules.models.ScaleShiftMACE",
• "__torch__.mace.modules.irreps_tools.reshape_irreps",
• "__torch__.torch.fx.graph_module.___torch_mangle_15.tp_forward",
• "__torch__.torch.fx.graph_module.___torch_mangle_9.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_8.GraphModule",
• "__torch__.e3nn.o3._linear.___torch_mangle_10.Linear",
• "__torch__.e3nn.o3._linear.___torch_mangle_55.Linear",
• "__torch__.e3nn.nn._fc.___torch_mangle_14._Layer",
• "__torch__.e3nn.o3._linear.___torch_mangle_22.Linear",
• "__torch__.torch.nn.modules.container.ParameterList",
• "torch.jit._pickle.restore_type_tag",
• "__torch__.mace.modules.blocks.ScaleShiftBlock",
• "__torch__.e3nn.o3._linear.___torch_mangle_42.Linear",
• "__torch__.mace.modules.radial.PolynomialCutoff",
• "__torch__.torch.fx.graph_module.___torch_mangle_36.GraphModule",
• "__torch__.e3nn.o3._tensor_product._tensor_product.TensorProduct",
• "torch.LongStorage",
• "__torch__.e3nn.math._normalize_activation.normalize2mom",
• "__torch__.mace.modules.symmetric_contraction.___torch_mangle_50.Contraction",
• "__torch__.torch.nn.modules.container.___torch_mangle_63.ModuleList",
• "__torch__.mace.modules.blocks.EquivariantProductBasisBlock",
• "__torch__.e3nn.nn._fc.___torch_mangle_21.FullyConnectedNet",
• "__torch__.mace.modules.blocks.RealAgnosticResidualInteractionBlock",
• "__torch__.torch.fx.graph_module.___torch_mangle_34.GraphModule",
• "__torch__.e3nn.math._normalize_activation.___torch_mangle_2.normalize2mom",
• "__torch__.mace.modules.blocks.LinearReadoutBlock",
• "__torch__.torch.fx.graph_module.___torch_mangle_5.GraphModule",
• "__torch__.torch.nn.modules.container.___torch_mangle_39.ModuleList",
• "__torch__.mace.modules.blocks.NonLinearReadoutBlock",
• "__torch__.e3nn.math._normalize_activation.___torch_mangle_15.normalize2mom",
• "__torch__.torch.fx.graph_module.___torch_mangle_28.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_35.GraphModule",
• "__torch__.torch.nn.modules.container.___torch_mangle_51.ModuleList",
• "__torch__.mace.modules.blocks.AtomicEnergiesBlock",
• "__torch__.e3nn.nn._fc.FullyConnectedNet",
• "__torch__.torch.fx.graph_module.___torch_mangle_37.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_4.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_46.GraphModule",
• "__torch__.mace.modules.symmetric_contraction.___torch_mangle_40.Contraction",
• "__torch__.torch.fx.graph_module.___torch_mangle_3.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_17.GraphModule",
• "__torch__.torch.nn.modules.container.___torch_mangle_41.ModuleList",
• "__torch__.e3nn.math._normalize_activation.___torch_mangle_5.normalize2mom",
• "__torch__.e3nn.o3._tensor_product._sub.FullyConnectedTensorProduct",
• "__torch__.torch.fx.graph_module.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_44.GraphModule",
• "__torch__.torch.nn.modules.container.___torch_mangle_59.ModuleList",
• "__torch__.torch.fx.graph_module.___torch_mangle_16.GraphModule",
• "__torch__.mace.modules.symmetric_contraction.SymmetricContraction",
• "__torch__.mace.modules.blocks.RadialEmbeddingBlock",
• "__torch__.torch.fx.graph_module.___torch_mangle_30.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_47.GraphModule",
• "__torch__.e3nn.math._normalize_activation.___torch_mangle_12.normalize2mom",
• "__torch__.mace.modules.blocks.LinearNodeEmbeddingBlock",
• "__torch__.torch.fx.graph_module.___torch_mangle_29.GraphModule",
• "__torch__.e3nn.nn._fc.___torch_mangle_4._Layer",
• "__torch__.e3nn.o3._spherical_harmonics.SphericalHarmonics",
• "__torch__.torch.nn.modules.container.___torch_mangle_48.ModuleList",
• "__torch__.e3nn.nn._fc.___torch_mangle_20._Layer",
• "__torch__.torch.fx.graph_module.___torch_mangle_33.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_18.GraphModule",
• "__torch__.e3nn.o3._linear.___torch_mangle_56.Linear",
• "__torch__.torch.fx.graph_module.___torch_mangle_14.GraphModule",
• "__torch__.mace.modules.symmetric_contraction.___torch_mangle_52.SymmetricContraction",
• "__torch__.e3nn.o3._tensor_product._sub.___torch_mangle_23.FullyConnectedTensorProduct",
• "__torch__.torch.fx.graph_module.___torch_mangle_6.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_10.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_45.GraphModule",
• "__torch__.e3nn.o3._linear.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_65.ModuleList",
• "__torch__.torch.nn.modules.container.___torch_mangle_32.ModuleList",
• "torch.jit._pickle.build_intlist",
• "__torch__.mace.calculators.lammps_mace.LAMMPS_MACE",
• "__torch__.e3nn.o3._linear.___torch_mangle_53.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_38.ModuleList",
• "__torch__.e3nn.o3._tensor_product._tensor_product.___torch_mangle_11.TensorProduct",
• "__torch__.e3nn.nn._fc.___torch_mangle_8._Layer",
• "__torch__.e3nn.o3._linear.___torch_mangle_9.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_64.ModuleList",
• "__torch__.torch.fx.graph_module.___torch_mangle_11.GraphModule",
• "__torch__.mace.modules.symmetric_contraction.Contraction",
• "__torch__.e3nn.nn._activation.Activation",
• "__torch__.torch.fx.graph_module.___torch_mangle_43.GraphModule",
• "__torch__.e3nn.nn._fc.___torch_mangle_7._Layer",
• "torch._utils._rebuild_tensor_v2",
• "collections.OrderedDict",
• "torch.DoubleStorage"

How to fix it?