rschirem

/

350K-small-dataset

like 0

• "__torch__.torch.fx.graph_module.___torch_mangle_11.GraphModule",
• "torch.jit._pickle.restore_type_tag",
• "torch.jit._pickle.build_intlist",
• "__torch__.torch.nn.modules.container.___torch_mangle_23.ModuleList",
• "__torch__.torch.fx.graph_module.tp_forward",
• "__torch__.nequip.nn._grad_output.StressOutput",
• "__torch__.torch.fx.graph_module.___torch_mangle_7.GraphModule",
• "__torch__.allegro.nn._fc.___torch_mangle_26.ScalarMLPFunction",
• "__torch__.torch.nn.modules.container.___torch_mangle_22.ModuleList",
• "__torch__.torch.fx.graph_module.___torch_mangle_6.tp_forward",
• "__torch__.nequip.nn._atomwise.PerSpeciesScaleShift",
• "__torch__.allegro.nn._fc.___torch_mangle_25.ScalarMLPFunction",
• "__torch__.allegro.nn._fc.ScalarMLP",
• "__torch__.allegro.nn._fc.___torch_mangle_21.ScalarMLPFunction",
• "__torch__.allegro.nn._edgewise.EdgewiseEnergySum",
• "__torch__.allegro.nn._strided._channels.MakeWeightedChannels",
• "__torch__.torch.fx.graph_module.___torch_mangle_9.GraphModule",
• "__torch__.allegro.nn._fc.___torch_mangle_16.ScalarMLPFunction",
• "__torch__.e3nn.o3._spherical_harmonics.SphericalHarmonics",
• "__torch__.allegro.nn._fc.___torch_mangle_18.ScalarMLPFunction",
• "__torch__.allegro.nn._allegro.Allegro_Module",
• "__torch__.torch.nn.modules.container.___torch_mangle_24.ModuleList",
• "__torch__.torch.fx.graph_module.GraphModule",
• "torch.FloatStorage",
• "__torch__.torch.fx.graph_module.___torch_mangle_14.GraphModule",
• "__torch__.nequip.nn._graph_mixin.SequentialGraphNetwork",
• "__torch__.torch.fx.graph_module.___torch_mangle_12.GraphModule",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.nequip.nn._graph_model.GraphModel",
• "__torch__.allegro.nn._fc.___torch_mangle_17.ScalarMLPFunction",
• "__torch__.nequip.nn.embedding._edge.SphericalHarmonicEdgeAttrs",
• "__torch__.torch.fx.graph_module.___torch_mangle_8.GraphModule",
• "__torch__.torch.nn.modules.container.ModuleList",
• "__torch__.torch.fx.graph_module.___torch_mangle_5.GraphModule",
• "__torch__.nequip.nn._rescale.RescaleOutput",
• "__torch__.torch.fx.graph_module.___torch_mangle_13.GraphModule",
• "__torch__.allegro.nn._fc.___torch_mangle_20.ScalarMLPFunction",
• "collections.OrderedDict",
• "__torch__.allegro.nn._fc.___torch_mangle_19.ScalarMLPFunction",
• "__torch__.torch.fx.graph_module.___torch_mangle_4.GraphModule",
• "__torch__.torch.fx.graph_module.___torch_mangle_15.GraphModule",
• "__torch__.allegro.nn._edgeembed.ProductTypeEmbedding",
• "__torch__.allegro.nn._fc.ScalarMLPFunction",
• "__torch__.nequip.nn._atomwise.AtomwiseReduce",
• "__torch__.torch.fx.graph_module.___torch_mangle_10.tp_forward",
• "__torch__.torch.fx.graph_module.___torch_mangle_3.GraphModule",
• "__torch__.allegro.nn._radial_basis.AllegroBesselBasis"

How to fix it?