NMThuan032k

/

mini_grp_q1

like 0

• "torch.nn.modules.sparse.Embedding",
• "collections.OrderedDict",
• "collections.defaultdict",
• "grp_model.GRP",
• "grp_model.Block",
• "omegaconf.nodes.AnyNode",
• "omegaconf.listconfig.ListConfig",
• "torch.nn.modules.linear.Linear",
• "torch.FloatStorage",
• "torch.nn.modules.container.Sequential",
• "grp_model.MultiHeadAttention",
• "torch.nn.modules.dropout.Dropout",
• "dill._dill._load_type",
• "grp_model.FeedFoward",
• "omegaconf.base.ContainerMetadata",
• "omegaconf.base.Metadata",
• "torch.nn.modules.container.ModuleList",
• "torch.nn.modules.activation.ReLU",
• "torch.nn.modules.normalization.LayerNorm",
• "omegaconf.dictconfig.DictConfig",
• "torch._utils._rebuild_parameter",
• "torch._utils._rebuild_tensor_v2",
• "typing.Any",
• "grp_model.Head"

How to fix it?