CSML-IIT

/

encoderops

like 0

Follow

Computational Statistics and Machine Learning unit @IIT 6

• "torch.optim.adamw.AdamW",
• "lightning.pytorch.loops.progress._ProcessedTracker",
• "torch.float32",
• "numpy.ndarray",
• "torchmetrics.metric.jit_distributed_available",
• "kooplearn.nn.losses.VAMPLoss",
• "functools.partial",
• "lightning.pytorch.callbacks.timer.Interval",
• "lightning.pytorch.utilities.model_helpers._ModuleMode",
• "lightning.pytorch.trainer.connectors.signal_connector._SignalConnector",
• "lightning.pytorch.trainer.connectors.callback_connector._CallbackConnector",
• "linear_operator_learning.nn.modules.mlp.MLP",
• "lightning.pytorch.strategies.strategy._ForwardRedirection",
• "lightning.pytorch.callbacks.progress.tqdm_progress.TQDMProgressBar",
• "numpy.core.multiarray._reconstruct",
• "lightning.pytorch.loops.optimization.manual._ManualOptimization",
• "kooplearn.nn.data.collate_context_dataset",
• "torch.device",
• "lightning.pytorch.loops.progress._ReadyCompletedTracker",
• "torch.utils.data.dataloader.DataLoader",
• "lightning.pytorch.trainer.states.TrainerState",
• "lightning.pytorch.trainer.connectors.data_connector._DataLoaderSource",
• "lightning.pytorch.trainer.states.TrainerStatus",
• "lightning.pytorch.trainer.connectors.accelerator_connector._AcceleratorConnector",
• "lightning.pytorch.loops.optimization.automatic._AutomaticOptimization",
• "builtins.getattr",
• "torch.nn.modules.activation.ReLU",
• "lightning.pytorch.trainer.connectors.data_connector._DataConnector",
• "kooplearn.models.feature_maps.nn.PLModule",
• "torch._C._VariableFunctionsClass",
• "fsspec.implementations.local.LocalFileSystem",
• "lightning.pytorch.accelerators.cuda.CUDAAccelerator",
• "__main__.MLPEncoder",
• "lightning.pytorch.callbacks.model_summary.ModelSummary",
• "lightning.pytorch.trainer.states.TrainerFn",
• "torch._utils._rebuild_parameter",
• "lightning.pytorch.profilers.base.PassThroughProfiler",
• "torch.nn.modules.linear.Linear",
• "torch.utils.data.sampler.BatchSampler",
• "lightning.pytorch.loggers.wandb.WandbLogger",
• "torch.nn.modules.container.Sequential",
• "lightning.pytorch.loops.fit_loop._FitLoop",
• "lightning.fabric.utilities.data.AttributeDict",
• "lightning.pytorch.trainer.connectors.checkpoint_connector._CheckpointConnector",
• "lightning.pytorch.trainer.connectors.logger_connector.logger_connector._LoggerConnector",
• "builtins.int",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync.no_op",
• "torch.utils.data.sampler.RandomSampler",
• "lightning.pytorch.loops.progress._Progress",
• "kooplearn.data.TrajectoryContextDataset",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultMetric",
• "numpy.dtype",
• "lightning.fabric.plugins.environments.lightning.LightningEnvironment",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Metadata",
• "lightning.pytorch.trainer.connectors.data_connector._DataHookSelector",
• "torch.utils._pytree.LeafSpec",
• "lightning.pytorch.loops.progress._SchedulerProgress",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync",
• "lightning.pytorch.loops.prediction_loop._PredictionLoop",
• "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO",
• "lightning.pytorch.loops.progress._OptimizationProgress",
• "torch.nn.modules.dropout.Dropout",
• "kooplearn.models.feature_maps.nn.NNFeatureMap",
• "kooplearn.data.TensorContextDataset",
• "lightning.pytorch.loops.evaluation_loop._EvaluationLoop",
• "lightning.pytorch.callbacks.model_checkpoint.ModelCheckpoint",
• "lightning_utilities.core.rank_zero.WarningCache",
• "linear_operator_learning.nn.modules.mlp._MLPBlock",
• "builtins.dict",
• "lightning.pytorch.trainer.states.RunningStage",
• "lightning.pytorch.utilities.combined_loader.CombinedLoader",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultCollection",
• "lightning.pytorch.loops.progress._StartedTracker",
• "lightning.pytorch.trainer.trainer.Trainer",
• "collections.defaultdict",
• "lightning.pytorch.plugins.precision.precision.Precision",
• "lightning.pytorch.loops.progress._OptimizerProgress",
• "lightning.pytorch.loops.progress._BatchProgress",
• "torch.storage._load_from_bytes",
• "collections.OrderedDict",
• "numpy.core.numeric._frombuffer",
• "torch._utils._rebuild_tensor_v2",
• "lightning.pytorch.strategies.single_device.SingleDeviceStrategy",
• "lightning.pytorch.callbacks.timer.Timer",
• "lightning.pytorch.loops.training_epoch_loop._TrainingEpochLoop",
• "fsspec.spec.make_instance"

How to fix it?

• "torch.optim.adamw.AdamW",
• "lightning.pytorch.loops.progress._ProcessedTracker",
• "torch.float32",
• "numpy.ndarray",
• "torchmetrics.metric.jit_distributed_available",
• "kooplearn.nn.losses.VAMPLoss",
• "functools.partial",
• "lightning.pytorch.callbacks.timer.Interval",
• "lightning.pytorch.utilities.model_helpers._ModuleMode",
• "lightning.pytorch.trainer.connectors.signal_connector._SignalConnector",
• "lightning.pytorch.trainer.connectors.callback_connector._CallbackConnector",
• "linear_operator_learning.nn.modules.mlp.MLP",
• "lightning.pytorch.strategies.strategy._ForwardRedirection",
• "lightning.pytorch.callbacks.progress.tqdm_progress.TQDMProgressBar",
• "numpy.core.multiarray._reconstruct",
• "lightning.pytorch.loops.optimization.manual._ManualOptimization",
• "kooplearn.nn.data.collate_context_dataset",
• "torch.device",
• "lightning.pytorch.loops.progress._ReadyCompletedTracker",
• "torch.utils.data.dataloader.DataLoader",
• "lightning.pytorch.trainer.states.TrainerState",
• "lightning.pytorch.trainer.connectors.data_connector._DataLoaderSource",
• "lightning.pytorch.trainer.states.TrainerStatus",
• "lightning.pytorch.trainer.connectors.accelerator_connector._AcceleratorConnector",
• "lightning.pytorch.loops.optimization.automatic._AutomaticOptimization",
• "builtins.getattr",
• "torch.nn.modules.activation.ReLU",
• "lightning.pytorch.trainer.connectors.data_connector._DataConnector",
• "kooplearn.models.feature_maps.nn.PLModule",
• "torch._C._VariableFunctionsClass",
• "fsspec.implementations.local.LocalFileSystem",
• "lightning.pytorch.accelerators.cuda.CUDAAccelerator",
• "__main__.MLPEncoder",
• "lightning.pytorch.callbacks.model_summary.ModelSummary",
• "lightning.pytorch.trainer.states.TrainerFn",
• "torch._utils._rebuild_parameter",
• "lightning.pytorch.profilers.base.PassThroughProfiler",
• "torch.nn.modules.linear.Linear",
• "torch.utils.data.sampler.BatchSampler",
• "lightning.pytorch.loggers.wandb.WandbLogger",
• "torch.nn.modules.container.Sequential",
• "lightning.pytorch.loops.fit_loop._FitLoop",
• "lightning.fabric.utilities.data.AttributeDict",
• "lightning.pytorch.trainer.connectors.checkpoint_connector._CheckpointConnector",
• "lightning.pytorch.trainer.connectors.logger_connector.logger_connector._LoggerConnector",
• "builtins.int",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync.no_op",
• "torch.utils.data.sampler.RandomSampler",
• "lightning.pytorch.loops.progress._Progress",
• "kooplearn.data.TrajectoryContextDataset",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultMetric",
• "numpy.dtype",
• "lightning.fabric.plugins.environments.lightning.LightningEnvironment",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Metadata",
• "lightning.pytorch.trainer.connectors.data_connector._DataHookSelector",
• "torch.utils._pytree.LeafSpec",
• "lightning.pytorch.loops.progress._SchedulerProgress",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync",
• "lightning.pytorch.loops.prediction_loop._PredictionLoop",
• "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO",
• "lightning.pytorch.loops.progress._OptimizationProgress",
• "torch.nn.modules.dropout.Dropout",
• "kooplearn.models.feature_maps.nn.NNFeatureMap",
• "kooplearn.data.TensorContextDataset",
• "lightning.pytorch.loops.evaluation_loop._EvaluationLoop",
• "lightning.pytorch.callbacks.model_checkpoint.ModelCheckpoint",
• "lightning_utilities.core.rank_zero.WarningCache",
• "linear_operator_learning.nn.modules.mlp._MLPBlock",
• "builtins.dict",
• "lightning.pytorch.trainer.states.RunningStage",
• "lightning.pytorch.utilities.combined_loader.CombinedLoader",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultCollection",
• "lightning.pytorch.loops.progress._StartedTracker",
• "lightning.pytorch.trainer.trainer.Trainer",
• "collections.defaultdict",
• "lightning.pytorch.plugins.precision.precision.Precision",
• "lightning.pytorch.loops.progress._OptimizerProgress",
• "lightning.pytorch.loops.progress._BatchProgress",
• "torch.storage._load_from_bytes",
• "collections.OrderedDict",
• "numpy.core.numeric._frombuffer",
• "torch._utils._rebuild_tensor_v2",
• "lightning.pytorch.strategies.single_device.SingleDeviceStrategy",
• "lightning.pytorch.callbacks.timer.Timer",
• "lightning.pytorch.loops.training_epoch_loop._TrainingEpochLoop",
• "fsspec.spec.make_instance"

How to fix it?

• "torch.optim.adamw.AdamW",
• "lightning.pytorch.loops.progress._ProcessedTracker",
• "torch.float32",
• "numpy.ndarray",
• "torchmetrics.metric.jit_distributed_available",
• "kooplearn.nn.losses.VAMPLoss",
• "functools.partial",
• "lightning.pytorch.callbacks.timer.Interval",
• "lightning.pytorch.utilities.model_helpers._ModuleMode",
• "lightning.pytorch.trainer.connectors.signal_connector._SignalConnector",
• "lightning.pytorch.trainer.connectors.callback_connector._CallbackConnector",
• "linear_operator_learning.nn.modules.mlp.MLP",
• "lightning.pytorch.strategies.strategy._ForwardRedirection",
• "lightning.pytorch.callbacks.progress.tqdm_progress.TQDMProgressBar",
• "numpy.core.multiarray._reconstruct",
• "lightning.pytorch.loops.optimization.manual._ManualOptimization",
• "kooplearn.nn.data.collate_context_dataset",
• "torch.device",
• "lightning.pytorch.loops.progress._ReadyCompletedTracker",
• "torch.utils.data.dataloader.DataLoader",
• "lightning.pytorch.trainer.states.TrainerState",
• "lightning.pytorch.trainer.connectors.data_connector._DataLoaderSource",
• "lightning.pytorch.trainer.states.TrainerStatus",
• "lightning.pytorch.trainer.connectors.accelerator_connector._AcceleratorConnector",
• "lightning.pytorch.loops.optimization.automatic._AutomaticOptimization",
• "builtins.getattr",
• "torch.nn.modules.activation.ReLU",
• "lightning.pytorch.trainer.connectors.data_connector._DataConnector",
• "kooplearn.models.feature_maps.nn.PLModule",
• "torch._C._VariableFunctionsClass",
• "fsspec.implementations.local.LocalFileSystem",
• "lightning.pytorch.accelerators.cuda.CUDAAccelerator",
• "__main__.MLPEncoder",
• "lightning.pytorch.callbacks.model_summary.ModelSummary",
• "lightning.pytorch.trainer.states.TrainerFn",
• "torch._utils._rebuild_parameter",
• "lightning.pytorch.profilers.base.PassThroughProfiler",
• "torch.nn.modules.linear.Linear",
• "torch.utils.data.sampler.BatchSampler",
• "lightning.pytorch.loggers.wandb.WandbLogger",
• "torch.nn.modules.container.Sequential",
• "lightning.pytorch.loops.fit_loop._FitLoop",
• "lightning.fabric.utilities.data.AttributeDict",
• "lightning.pytorch.trainer.connectors.checkpoint_connector._CheckpointConnector",
• "lightning.pytorch.trainer.connectors.logger_connector.logger_connector._LoggerConnector",
• "builtins.int",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync.no_op",
• "torch.utils.data.sampler.RandomSampler",
• "lightning.pytorch.loops.progress._Progress",
• "kooplearn.data.TrajectoryContextDataset",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultMetric",
• "numpy.dtype",
• "lightning.fabric.plugins.environments.lightning.LightningEnvironment",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Metadata",
• "lightning.pytorch.trainer.connectors.data_connector._DataHookSelector",
• "torch.utils._pytree.LeafSpec",
• "lightning.pytorch.loops.progress._SchedulerProgress",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync",
• "lightning.pytorch.loops.prediction_loop._PredictionLoop",
• "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO",
• "lightning.pytorch.loops.progress._OptimizationProgress",
• "torch.nn.modules.dropout.Dropout",
• "kooplearn.models.feature_maps.nn.NNFeatureMap",
• "kooplearn.data.TensorContextDataset",
• "lightning.pytorch.loops.evaluation_loop._EvaluationLoop",
• "lightning.pytorch.callbacks.model_checkpoint.ModelCheckpoint",
• "lightning_utilities.core.rank_zero.WarningCache",
• "linear_operator_learning.nn.modules.mlp._MLPBlock",
• "builtins.dict",
• "lightning.pytorch.trainer.states.RunningStage",
• "lightning.pytorch.utilities.combined_loader.CombinedLoader",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultCollection",
• "lightning.pytorch.loops.progress._StartedTracker",
• "lightning.pytorch.trainer.trainer.Trainer",
• "collections.defaultdict",
• "lightning.pytorch.plugins.precision.precision.Precision",
• "lightning.pytorch.loops.progress._OptimizerProgress",
• "lightning.pytorch.loops.progress._BatchProgress",
• "torch.storage._load_from_bytes",
• "collections.OrderedDict",
• "numpy.core.numeric._frombuffer",
• "torch._utils._rebuild_tensor_v2",
• "lightning.pytorch.strategies.single_device.SingleDeviceStrategy",
• "lightning.pytorch.callbacks.timer.Timer",
• "lightning.pytorch.loops.training_epoch_loop._TrainingEpochLoop",
• "fsspec.spec.make_instance"

How to fix it?

• "torch.optim.adamw.AdamW",
• "lightning.pytorch.loops.progress._ProcessedTracker",
• "torch.float32",
• "numpy.ndarray",
• "torchmetrics.metric.jit_distributed_available",
• "kooplearn.nn.losses.VAMPLoss",
• "functools.partial",
• "lightning.pytorch.callbacks.timer.Interval",
• "lightning.pytorch.utilities.model_helpers._ModuleMode",
• "lightning.pytorch.trainer.connectors.signal_connector._SignalConnector",
• "lightning.pytorch.trainer.connectors.callback_connector._CallbackConnector",
• "linear_operator_learning.nn.modules.mlp.MLP",
• "lightning.pytorch.strategies.strategy._ForwardRedirection",
• "lightning.pytorch.callbacks.progress.tqdm_progress.TQDMProgressBar",
• "numpy.core.multiarray._reconstruct",
• "lightning.pytorch.loops.optimization.manual._ManualOptimization",
• "kooplearn.nn.data.collate_context_dataset",
• "torch.device",
• "lightning.pytorch.loops.progress._ReadyCompletedTracker",
• "torch.utils.data.dataloader.DataLoader",
• "lightning.pytorch.trainer.states.TrainerState",
• "lightning.pytorch.trainer.connectors.data_connector._DataLoaderSource",
• "lightning.pytorch.trainer.states.TrainerStatus",
• "lightning.pytorch.trainer.connectors.accelerator_connector._AcceleratorConnector",
• "lightning.pytorch.loops.optimization.automatic._AutomaticOptimization",
• "builtins.getattr",
• "torch.nn.modules.activation.ReLU",
• "lightning.pytorch.trainer.connectors.data_connector._DataConnector",
• "kooplearn.models.feature_maps.nn.PLModule",
• "torch._C._VariableFunctionsClass",
• "fsspec.implementations.local.LocalFileSystem",
• "lightning.pytorch.accelerators.cuda.CUDAAccelerator",
• "__main__.MLPEncoder",
• "lightning.pytorch.callbacks.model_summary.ModelSummary",
• "lightning.pytorch.trainer.states.TrainerFn",
• "torch._utils._rebuild_parameter",
• "lightning.pytorch.profilers.base.PassThroughProfiler",
• "torch.nn.modules.linear.Linear",
• "torch.utils.data.sampler.BatchSampler",
• "lightning.pytorch.loggers.wandb.WandbLogger",
• "torch.nn.modules.container.Sequential",
• "lightning.pytorch.loops.fit_loop._FitLoop",
• "lightning.fabric.utilities.data.AttributeDict",
• "lightning.pytorch.trainer.connectors.checkpoint_connector._CheckpointConnector",
• "lightning.pytorch.trainer.connectors.logger_connector.logger_connector._LoggerConnector",
• "builtins.int",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync.no_op",
• "torch.utils.data.sampler.RandomSampler",
• "lightning.pytorch.loops.progress._Progress",
• "kooplearn.data.TrajectoryContextDataset",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultMetric",
• "numpy.dtype",
• "lightning.fabric.plugins.environments.lightning.LightningEnvironment",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Metadata",
• "lightning.pytorch.trainer.connectors.data_connector._DataHookSelector",
• "torch.utils._pytree.LeafSpec",
• "lightning.pytorch.loops.progress._SchedulerProgress",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync",
• "lightning.pytorch.loops.prediction_loop._PredictionLoop",
• "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO",
• "lightning.pytorch.loops.progress._OptimizationProgress",
• "torch.nn.modules.dropout.Dropout",
• "kooplearn.models.feature_maps.nn.NNFeatureMap",
• "kooplearn.data.TensorContextDataset",
• "lightning.pytorch.loops.evaluation_loop._EvaluationLoop",
• "lightning.pytorch.callbacks.model_checkpoint.ModelCheckpoint",
• "lightning_utilities.core.rank_zero.WarningCache",
• "linear_operator_learning.nn.modules.mlp._MLPBlock",
• "builtins.dict",
• "lightning.pytorch.trainer.states.RunningStage",
• "lightning.pytorch.utilities.combined_loader.CombinedLoader",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultCollection",
• "lightning.pytorch.loops.progress._StartedTracker",
• "lightning.pytorch.trainer.trainer.Trainer",
• "collections.defaultdict",
• "lightning.pytorch.plugins.precision.precision.Precision",
• "lightning.pytorch.loops.progress._OptimizerProgress",
• "lightning.pytorch.loops.progress._BatchProgress",
• "torch.storage._load_from_bytes",
• "collections.OrderedDict",
• "numpy.core.numeric._frombuffer",
• "torch._utils._rebuild_tensor_v2",
• "lightning.pytorch.strategies.single_device.SingleDeviceStrategy",
• "lightning.pytorch.callbacks.timer.Timer",
• "lightning.pytorch.loops.training_epoch_loop._TrainingEpochLoop",
• "fsspec.spec.make_instance"

How to fix it?

• "torch.optim.adamw.AdamW",
• "lightning.pytorch.loops.progress._ProcessedTracker",
• "torch.float32",
• "numpy.ndarray",
• "torchmetrics.metric.jit_distributed_available",
• "kooplearn.nn.losses.VAMPLoss",
• "functools.partial",
• "lightning.pytorch.callbacks.timer.Interval",
• "lightning.pytorch.utilities.model_helpers._ModuleMode",
• "lightning.pytorch.trainer.connectors.signal_connector._SignalConnector",
• "lightning.pytorch.trainer.connectors.callback_connector._CallbackConnector",
• "linear_operator_learning.nn.modules.mlp.MLP",
• "lightning.pytorch.strategies.strategy._ForwardRedirection",
• "lightning.pytorch.callbacks.progress.tqdm_progress.TQDMProgressBar",
• "numpy.core.multiarray._reconstruct",
• "lightning.pytorch.loops.optimization.manual._ManualOptimization",
• "kooplearn.nn.data.collate_context_dataset",
• "torch.device",
• "lightning.pytorch.loops.progress._ReadyCompletedTracker",
• "torch.utils.data.dataloader.DataLoader",
• "lightning.pytorch.trainer.states.TrainerState",
• "lightning.pytorch.trainer.connectors.data_connector._DataLoaderSource",
• "lightning.pytorch.trainer.states.TrainerStatus",
• "lightning.pytorch.trainer.connectors.accelerator_connector._AcceleratorConnector",
• "lightning.pytorch.loops.optimization.automatic._AutomaticOptimization",
• "builtins.getattr",
• "torch.nn.modules.activation.ReLU",
• "lightning.pytorch.trainer.connectors.data_connector._DataConnector",
• "kooplearn.models.feature_maps.nn.PLModule",
• "torch._C._VariableFunctionsClass",
• "fsspec.implementations.local.LocalFileSystem",
• "lightning.pytorch.accelerators.cuda.CUDAAccelerator",
• "__main__.MLPEncoder",
• "lightning.pytorch.callbacks.model_summary.ModelSummary",
• "lightning.pytorch.trainer.states.TrainerFn",
• "torch._utils._rebuild_parameter",
• "lightning.pytorch.profilers.base.PassThroughProfiler",
• "torch.nn.modules.linear.Linear",
• "torch.utils.data.sampler.BatchSampler",
• "lightning.pytorch.loggers.wandb.WandbLogger",
• "torch.nn.modules.container.Sequential",
• "lightning.pytorch.loops.fit_loop._FitLoop",
• "lightning.fabric.utilities.data.AttributeDict",
• "lightning.pytorch.trainer.connectors.checkpoint_connector._CheckpointConnector",
• "lightning.pytorch.trainer.connectors.logger_connector.logger_connector._LoggerConnector",
• "builtins.int",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync.no_op",
• "torch.utils.data.sampler.RandomSampler",
• "lightning.pytorch.loops.progress._Progress",
• "kooplearn.data.TrajectoryContextDataset",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultMetric",
• "numpy.dtype",
• "lightning.fabric.plugins.environments.lightning.LightningEnvironment",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Metadata",
• "lightning.pytorch.trainer.connectors.data_connector._DataHookSelector",
• "torch.utils._pytree.LeafSpec",
• "lightning.pytorch.loops.progress._SchedulerProgress",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync",
• "lightning.pytorch.loops.prediction_loop._PredictionLoop",
• "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO",
• "lightning.pytorch.loops.progress._OptimizationProgress",
• "torch.nn.modules.dropout.Dropout",
• "kooplearn.models.feature_maps.nn.NNFeatureMap",
• "kooplearn.data.TensorContextDataset",
• "lightning.pytorch.loops.evaluation_loop._EvaluationLoop",
• "lightning.pytorch.callbacks.model_checkpoint.ModelCheckpoint",
• "lightning_utilities.core.rank_zero.WarningCache",
• "linear_operator_learning.nn.modules.mlp._MLPBlock",
• "builtins.dict",
• "lightning.pytorch.trainer.states.RunningStage",
• "lightning.pytorch.utilities.combined_loader.CombinedLoader",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultCollection",
• "lightning.pytorch.loops.progress._StartedTracker",
• "lightning.pytorch.trainer.trainer.Trainer",
• "collections.defaultdict",
• "lightning.pytorch.plugins.precision.precision.Precision",
• "lightning.pytorch.loops.progress._OptimizerProgress",
• "lightning.pytorch.loops.progress._BatchProgress",
• "torch.storage._load_from_bytes",
• "collections.OrderedDict",
• "numpy.core.numeric._frombuffer",
• "torch._utils._rebuild_tensor_v2",
• "lightning.pytorch.strategies.single_device.SingleDeviceStrategy",
• "lightning.pytorch.callbacks.timer.Timer",
• "lightning.pytorch.loops.training_epoch_loop._TrainingEpochLoop",
• "fsspec.spec.make_instance"

How to fix it?

• "torch.optim.adamw.AdamW",
• "lightning.pytorch.loops.progress._ProcessedTracker",
• "torch.float32",
• "numpy.ndarray",
• "torchmetrics.metric.jit_distributed_available",
• "kooplearn.nn.losses.VAMPLoss",
• "functools.partial",
• "lightning.pytorch.callbacks.timer.Interval",
• "lightning.pytorch.utilities.model_helpers._ModuleMode",
• "lightning.pytorch.trainer.connectors.signal_connector._SignalConnector",
• "lightning.pytorch.trainer.connectors.callback_connector._CallbackConnector",
• "linear_operator_learning.nn.modules.mlp.MLP",
• "lightning.pytorch.strategies.strategy._ForwardRedirection",
• "lightning.pytorch.callbacks.progress.tqdm_progress.TQDMProgressBar",
• "numpy.core.multiarray._reconstruct",
• "lightning.pytorch.loops.optimization.manual._ManualOptimization",
• "kooplearn.nn.data.collate_context_dataset",
• "torch.device",
• "lightning.pytorch.loops.progress._ReadyCompletedTracker",
• "torch.utils.data.dataloader.DataLoader",
• "lightning.pytorch.trainer.states.TrainerState",
• "lightning.pytorch.trainer.connectors.data_connector._DataLoaderSource",
• "lightning.pytorch.trainer.states.TrainerStatus",
• "lightning.pytorch.trainer.connectors.accelerator_connector._AcceleratorConnector",
• "lightning.pytorch.loops.optimization.automatic._AutomaticOptimization",
• "builtins.getattr",
• "torch.nn.modules.activation.ReLU",
• "lightning.pytorch.trainer.connectors.data_connector._DataConnector",
• "kooplearn.models.feature_maps.nn.PLModule",
• "torch._C._VariableFunctionsClass",
• "fsspec.implementations.local.LocalFileSystem",
• "lightning.pytorch.accelerators.cuda.CUDAAccelerator",
• "__main__.MLPEncoder",
• "lightning.pytorch.callbacks.model_summary.ModelSummary",
• "lightning.pytorch.trainer.states.TrainerFn",
• "torch._utils._rebuild_parameter",
• "lightning.pytorch.profilers.base.PassThroughProfiler",
• "torch.nn.modules.linear.Linear",
• "torch.utils.data.sampler.BatchSampler",
• "lightning.pytorch.loggers.wandb.WandbLogger",
• "torch.nn.modules.container.Sequential",
• "lightning.pytorch.loops.fit_loop._FitLoop",
• "lightning.fabric.utilities.data.AttributeDict",
• "lightning.pytorch.trainer.connectors.checkpoint_connector._CheckpointConnector",
• "lightning.pytorch.trainer.connectors.logger_connector.logger_connector._LoggerConnector",
• "builtins.int",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync.no_op",
• "torch.utils.data.sampler.RandomSampler",
• "lightning.pytorch.loops.progress._Progress",
• "kooplearn.data.TrajectoryContextDataset",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultMetric",
• "numpy.dtype",
• "lightning.fabric.plugins.environments.lightning.LightningEnvironment",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Metadata",
• "lightning.pytorch.trainer.connectors.data_connector._DataHookSelector",
• "torch.utils._pytree.LeafSpec",
• "lightning.pytorch.loops.progress._SchedulerProgress",
• "lightning.pytorch.trainer.connectors.logger_connector.result._Sync",
• "lightning.pytorch.loops.prediction_loop._PredictionLoop",
• "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO",
• "lightning.pytorch.loops.progress._OptimizationProgress",
• "torch.nn.modules.dropout.Dropout",
• "kooplearn.models.feature_maps.nn.NNFeatureMap",
• "kooplearn.data.TensorContextDataset",
• "lightning.pytorch.loops.evaluation_loop._EvaluationLoop",
• "lightning.pytorch.callbacks.model_checkpoint.ModelCheckpoint",
• "lightning_utilities.core.rank_zero.WarningCache",
• "linear_operator_learning.nn.modules.mlp._MLPBlock",
• "builtins.dict",
• "lightning.pytorch.trainer.states.RunningStage",
• "lightning.pytorch.utilities.combined_loader.CombinedLoader",
• "lightning.pytorch.trainer.connectors.logger_connector.result._ResultCollection",
• "lightning.pytorch.loops.progress._StartedTracker",
• "lightning.pytorch.trainer.trainer.Trainer",
• "collections.defaultdict",
• "lightning.pytorch.plugins.precision.precision.Precision",
• "lightning.pytorch.loops.progress._OptimizerProgress",
• "lightning.pytorch.loops.progress._BatchProgress",
• "torch.storage._load_from_bytes",
• "collections.OrderedDict",
• "numpy.core.numeric._frombuffer",
• "torch._utils._rebuild_tensor_v2",
• "lightning.pytorch.strategies.single_device.SingleDeviceStrategy",
• "lightning.pytorch.callbacks.timer.Timer",
• "lightning.pytorch.loops.training_epoch_loop._TrainingEpochLoop",
• "fsspec.spec.make_instance"

How to fix it?