val-epoch=254-val_loss=0.4149.ckpt
Detected Pickle imports (57)
- "context_general_bci.config.config_base.DatasetConfig",
- "omegaconf.nodes.FloatNode",
- "__builtin__.unicode",
- "context_general_bci.config.config_base.PittConfig",
- "omegaconf.base.ContainerMetadata",
- "__builtin__.dict",
- "torch._utils._rebuild_tensor_v2",
- "context_general_bci.config.config_base.EmbedStrat",
- "typing.Any",
- "context_general_bci.config.config_base.ModelConfig",
- "omegaconf.nodes.StringNode",
- "omegaconf.dictconfig.DictConfig",
- "omegaconf.nodes.EnumNode",
- "context_general_bci.config.config_base.MazeConfig",
- "typing.List",
- "__builtin__.float",
- "context_general_bci.dataset.ContextAttrs",
- "__builtin__.bool",
- "context_general_bci.config.config_base.MetaKey",
- "__builtin__.list",
- "context_general_bci.config.config_base.TaskConfig",
- "omegaconf.nodes.PathNode",
- "omegaconf.base.Metadata",
- "context_general_bci.config.config_base.TrainConfig",
- "omegaconf.nodes.IntegerNode",
- "context_general_bci.config.config_base.TransformerConfig",
- "collections.OrderedDict",
- "context_general_bci.tasks.ExperimentalTask",
- "context_general_bci.config.config_base.EvalConfig",
- "context_general_bci.config.presets.JointDecodeFlatTaskConfigV2",
- "context_general_bci.subjects.SubjectName",
- "context_general_bci.config.config_base.FalconConfig",
- "omegaconf.nodes.AnyNode",
- "omegaconf.nodes.BooleanNode",
- "context_general_bci.config.config_base.ExperimentalConfig",
- "context_general_bci.config.config_base.Architecture",
- "context_general_bci.config.config_base.Output",
- "context_general_bci.config.config_base.RLConfig",
- "context_general_bci.config.presets.FlatEncDecTransformerConfig",
- "context_general_bci.config.config_base.RootConfig",
- "context_general_bci.config.config_base.DyerCOConfig",
- "__builtin__.long",
- "pathlib.PosixPath",
- "context_general_bci.config.config_base.NLBConfig",
- "context_general_bci.config.config_base.MillerConfig",
- "pathlib.Path",
- "context_general_bci.dataset.DataAttrs",
- "omegaconf.listconfig.ListConfig",
- "context_general_bci.config.config_base.DataKey",
- "context_general_bci.config.config_base.RTTConfig",
- "context_general_bci.config.presets.LargescaleModelConfig",
- "context_general_bci.config.presets.ScaleHistoryDatasetConfig",
- "collections.defaultdict",
- "context_general_bci.config.config_base.ModelTask",
- "context_general_bci.config.presets.LargescaleTrainConfig",
- "context_general_bci.config.config_base.Metric",
- "torch.FloatStorage"
How to fix it?
541 MB
init ckpts