add eval mode in sample
985a6a5
-
1.52 kB
initial commit
-
6.59 kB
add eval mode in sample
-
537 Bytes
update config with checkpointing
weights.ckpt
Detected Pickle imports (14)
- "omegaconf.dictconfig.DictConfig",
- "torch.LongStorage",
- "typing.Any",
- "__builtin__.dict",
- "omegaconf.nodes.AnyNode",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "omegaconf.base.Metadata",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.long",
- "__builtin__.list",
- "omegaconf.base.ContainerMetadata",
- "omegaconf.listconfig.ListConfig",
- "collections.defaultdict"
How to fix it?
271 MB
add weights and config for 1M model
