zap-thamm

/

Imitation-BC-CartPole-v1

like 0

• "torch.nn.modules.activation.Tanh",
• "torch.utils.data.sampler.BatchSampler",
• "stable_baselines3.common.torch_layers.MlpExtractor",
• "gymnasium.spaces.discrete.Discrete",
• "torch.storage._load_from_bytes",
• "torch._utils._rebuild_parameter",
• "numpy.random._pickle.__generator_ctor",
• "torch.nn.modules.container.Sequential",
• "numpy.ndarray",
• "torch._utils._rebuild_tensor_v2",
• "imitation.algorithms.bc.BC",
• "gymnasium.spaces.box.Box",
• "imitation.data.types.transitions_collate_fn",
• "collections.defaultdict",
• "numpy.core.multiarray.scalar",
• "numpy.core.multiarray._reconstruct",
• "numpy.dtype",
• "imitation.algorithms.bc.BehaviorCloningLossCalculator",
• "torch.nn.modules.linear.Linear",
• "collections.OrderedDict",
• "imitation.algorithms.bc.BCLogger",
• "torch.distributions.categorical.Categorical",
• "torch.optim.adam.Adam",
• "imitation.data.types.Transitions",
• "imitation.policies.base.FeedForward32Policy",
• "builtins.dict",
• "numpy.random._pickle.__bit_generator_ctor",
• "torch.utils.data.sampler.RandomSampler",
• "torch.Size",
• "stable_baselines3.common.distributions.CategoricalDistribution",
• "torch.utils.data.dataloader.DataLoader",
• "torch.nn.modules.flatten.Flatten",
• "stable_baselines3.common.torch_layers.FlattenExtractor"

How to fix it?