ameya98
/

JAMUN

molecular-dynamics

protein-folding

generative-models

diffusion-models

Model card Files Files and versions

JAMUN / checkpoints /MDGen_4AA

254 MB

2 contributors

History: 1 commit

ameya98's picture

Upload first set of checkpoints

ad25b72 verified 11 months ago

epoch=0-step=2730000.ckpt
Detected Pickle imports (20)
- "jamun.model.arch.e3conv.E3Conv",
- "jamun.e3tools.nn._conv.Conv",
- "jamun.e3tools.nn._conv.ConvBlock",
- "collections.defaultdict",
- "functools.partial",
- "torch.optim.adam.Adam",
- "__builtin__.dict",
- "omegaconf.listconfig.ListConfig",
- "typing.Any",
- "torch.FloatStorage",
- "jamun.e3tools.nn._mlp.EquivariantMLP",
- "omegaconf.nodes.AnyNode",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "omegaconf.dictconfig.DictConfig",
- "jamun.distributions._distributions.ConstantSigma",
- "__builtin__.long",
- "omegaconf.base.Metadata",
- "__builtin__.list",
- "omegaconf.base.ContainerMetadata"
How to fix it?
127 MB
xet

Upload first set of checkpoints 11 months ago
last.ckpt
Detected Pickle imports (20)
- "jamun.model.arch.e3conv.E3Conv",
- "jamun.e3tools.nn._conv.Conv",
- "jamun.e3tools.nn._conv.ConvBlock",
- "collections.defaultdict",
- "functools.partial",
- "torch.optim.adam.Adam",
- "__builtin__.dict",
- "omegaconf.listconfig.ListConfig",
- "typing.Any",
- "torch.FloatStorage",
- "jamun.e3tools.nn._mlp.EquivariantMLP",
- "omegaconf.nodes.AnyNode",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "omegaconf.dictconfig.DictConfig",
- "jamun.distributions._distributions.ConstantSigma",
- "__builtin__.long",
- "omegaconf.base.Metadata",
- "__builtin__.list",
- "omegaconf.base.ContainerMetadata"
How to fix it?
127 MB
xet

Upload first set of checkpoints 11 months ago