Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W | Z _ __init__() (campd.architectures.context.encoder.ContextEncoder method) (campd.architectures.context.encoder.KeyNetModule method) (campd.architectures.diffusion.temporal_unet.TemporalUnet method) (campd.architectures.layers.attention.BasicTransformerBlock method) (campd.architectures.layers.attention.CrossAttention method) (campd.architectures.layers.attention.FeedForward method) (campd.architectures.layers.attention.GEGLU method) (campd.architectures.layers.attention.SpatialTransformer method) (campd.architectures.layers.layers.Conv1dBlock method) (campd.architectures.layers.layers.Downsample1d method) (campd.architectures.layers.layers.LayerNorm method) (campd.architectures.layers.layers.MLP1D method) (campd.architectures.layers.layers.PreNorm method) (campd.architectures.layers.layers.Residual method) (campd.architectures.layers.layers.ResidualTemporalBlock method) (campd.architectures.layers.layers.SinusoidalPosEmb method) (campd.architectures.layers.layers.TimeEncoder method) (campd.architectures.layers.layers.Upsample1d method) (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) (campd.data.normalization.base.Normalizer method) (campd.data.normalization.normalization.DatasetNormalizer method) (campd.data.normalization.normalization.FixedLimitsNormalizer method) (campd.data.normalization.normalization.GaussianNormalizer method) (campd.data.normalization.normalization.Identity method) (campd.data.normalization.normalization.SafeLimitsNormalizer method) (campd.data.trajectory_dataset.TrajectoryDataset method) (campd.data.trajectory_sample.TrajectorySample method) (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) (campd.training.base.Trainer method) (campd.training.callbacks.CheckpointCallback method) (campd.training.callbacks.EarlyStoppingCallback method) (campd.training.callbacks.EMA method) (campd.training.callbacks.EMACallback method) (campd.training.callbacks.WandBCallback method) (campd.training.losses.WeightedLoss method) (campd.training.objectives.diffusion.DiffusionObjective method) (campd.training.summary.Summary method) (campd.utils.registry.FromCfg method) (campd.utils.registry.Registry method) (campd.utils.torch.TimerCUDA method) A act (campd.architectures.layers.layers.MLP1DCfg attribute) ACTIVATIONS (in module campd.architectures.layers.layers) add_noise() (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) add_time_emb_to_conditioning (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) all_keys() (campd.data.context.TrajectoryContext method) analytical_first_step (campd.models.diffusion.model.SamplingCfg attribute) analytical_first_step() (in module campd.models.helpers) append() (campd.data.embedded_context.EmbeddedContext method) apply_hard_conditioning() (in module campd.models.helpers) attention_dim_head (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) attention_num_heads (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) available() (campd.utils.registry.Registry method) B BasicTransformerBlock (class in campd.architectures.layers.attention) batch_size (campd.data.context.TrajectoryContext property) (campd.data.trajectory_sample.TrajectorySample property) (campd.models.diffusion.model.SamplingCfg attribute) (campd.training.base.CUDAGraphCfg attribute) build() (campd.utils.registry.Spec method) build_from() (campd.utils.registry.Spec method) C Callback (class in campd.training.callbacks) callbacks (campd.training.base.TrainerCfg attribute) campd.architectures.context.encoder module campd.architectures.diffusion.base module campd.architectures.diffusion.temporal_unet module campd.architectures.layers.attention module campd.architectures.layers.layers module campd.architectures.layers.utils module campd.data.context module campd.data.embedded_context module campd.data.normalization.base module campd.data.normalization.normalization module campd.data.trajectory_dataset module campd.data.trajectory_sample module campd.models.diffusion.model module campd.models.helpers module campd.runner module campd.training.base module campd.training.callbacks module campd.training.losses module campd.training.objectives.base module campd.training.objectives.diffusion module campd.training.summary module campd.utils.config module campd.utils.file module campd.utils.random module campd.utils.registry module campd.utils.stats module campd.utils.torch module CheckpointCallback (class in campd.training.callbacks) clip_grad (campd.training.base.TrainerCfg attribute) clone() (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) cls (campd.utils.registry.Spec attribute) collate() (campd.data.context.TrajectoryContext class method) (campd.data.trajectory_sample.TrajectorySample class method) components (campd.data.context.TrajectoryContext attribute) concat_batch() (campd.data.embedded_context.EmbeddedContext method) concat_config (campd.architectures.context.encoder.ContextEncoderCfg attribute) concat_keys() (campd.data.embedded_context.EmbeddedContext method) cond_scale (campd.models.diffusion.model.SamplingCfg attribute) conditioning_embed_dim (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) conditioning_key (campd.architectures.diffusion.temporal_unet.TemporalUnet attribute) config (campd.models.diffusion.model.ContextTrajectoryDiffusionModel attribute) (campd.utils.registry.Spec attribute) context (campd.data.trajectory_sample.TrajectorySample attribute) context_buffer_sizes (campd.models.diffusion.model.SamplingCfg attribute) (campd.training.base.CUDAGraphCfg attribute) context_dims (campd.architectures.context.encoder.ContextEncoder property) context_fields (campd.data.trajectory_dataset.HDF5FieldCfg attribute) context_keys (campd.architectures.context.encoder.ContextEncoder property) context_network (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) context_normalized (campd.data.trajectory_dataset.TrajectoryDataset attribute) ContextEncoder (class in campd.architectures.context.encoder) ContextEncoderCfg (class in campd.architectures.context.encoder) ContextTrajectoryDiffusionModel (class in campd.models.diffusion.model) ContextTrajectoryDiffusionModelCfg (class in campd.models.diffusion.model) Conv1dBlock (class in campd.architectures.layers.layers) convert_dict_to_sweep_dict() (in module campd.utils.config) copy_() (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) copy_from() (campd.data.context.TrajectoryContext method) CrossAttention (class in campd.architectures.layers.attention) cuda_graph (campd.training.base.TrainerCfg attribute) CUDAGraphCfg (class in campd.training.base) current_data (campd.training.base.Trainer attribute) current_epoch (campd.training.base.Trainer attribute) current_val_data (campd.training.base.Trainer attribute) D dataset_dir (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) DatasetNormalizer (class in campd.data.normalization.normalization) ddpm_scheduler_extra_kwargs (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) default() (in module campd.architectures.layers.utils) denormalize (campd.models.diffusion.model.SamplingCfg attribute) denormalize() (campd.data.context.TrajectoryContext method) (campd.data.trajectory_sample.TrajectorySample method) device (campd.utils.torch.TensorArgs attribute) DiffusionObjective (class in campd.training.objectives.diffusion) DiffusionObjectiveCfg (class in campd.training.objectives.diffusion) dim_mults (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) Downsample1d (class in campd.architectures.layers.layers) dtype (campd.utils.torch.TensorArgs attribute) E EarlyStoppingCallback (class in campd.training.callbacks) elapsed (campd.utils.torch.TimerCUDA property) EMA (class in campd.training.callbacks) ema_model (campd.training.base.Trainer attribute) EMACallback (class in campd.training.callbacks) EmbeddedContext (class in campd.data.embedded_context) embeddings (campd.data.embedded_context.EmbeddedContext attribute) enable_conditioning (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) enabled (campd.training.base.CUDAGraphCfg attribute) exists() (in module campd.architectures.layers.utils) expand() (campd.data.context.TrajectoryContext method) expand_shared_context() (campd.data.trajectory_sample.TrajectorySample method) expand_time() (in module campd.models.helpers) export() (campd.data.normalization.base.Normalizer method) export_config() (campd.data.normalization.normalization.DatasetNormalizer method) F FeedForward (class in campd.architectures.layers.attention) field_config (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) field_limits (campd.data.normalization.normalization.NormalizationCfg attribute) field_means (campd.data.normalization.normalization.NormalizationCfg attribute) field_stds (campd.data.normalization.normalization.NormalizationCfg attribute) FieldLimit (class in campd.data.normalization.normalization) fit() (campd.training.base.Trainer method) fix_random_seed() (in module campd.utils.random) FixedLimitsNormalizer (class in campd.data.normalization.normalization) flatten() (in module campd.data.normalization.normalization) forward() (campd.architectures.context.encoder.ContextEncoder method) (campd.architectures.diffusion.base.ReverseDiffusionNetwork method) (campd.architectures.diffusion.temporal_unet.TemporalUnet method) (campd.architectures.layers.attention.BasicTransformerBlock method) (campd.architectures.layers.attention.CrossAttention method) (campd.architectures.layers.attention.FeedForward method) (campd.architectures.layers.attention.GEGLU method) (campd.architectures.layers.attention.SpatialTransformer method) (campd.architectures.layers.layers.Conv1dBlock method) (campd.architectures.layers.layers.Downsample1d method) (campd.architectures.layers.layers.LayerNorm method) (campd.architectures.layers.layers.MLP1D method) (campd.architectures.layers.layers.PreNorm method) (campd.architectures.layers.layers.Residual method) (campd.architectures.layers.layers.ResidualTemporalBlock method) (campd.architectures.layers.layers.SinusoidalPosEmb method) (campd.architectures.layers.layers.TimeEncoder method) (campd.architectures.layers.layers.Upsample1d method) (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) (campd.training.losses.WeightedLoss method) forward_with_cond_scale() (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) freeze_torch_model_params() (in module campd.utils.torch) from_config() (campd.architectures.context.encoder.ContextEncoder class method) (campd.architectures.diffusion.temporal_unet.TemporalUnet class method) (campd.architectures.layers.layers.MLP1D class method) (campd.data.normalization.normalization.DatasetNormalizer class method) (campd.models.diffusion.model.ContextTrajectoryDiffusionModel class method) (campd.training.objectives.diffusion.DiffusionObjective class method) (campd.utils.registry.FromCfg class method) from_pretrained() (campd.models.diffusion.model.ContextTrajectoryDiffusionModel class method) FromCfg (class in campd.utils.registry) G GaussianNormalizer (class in campd.data.normalization.normalization) GEGLU (class in campd.architectures.layers.attention) get_context_fields() (campd.data.trajectory_dataset.HDF5FieldCfg method) get_dataloader() (campd.data.trajectory_dataset.TrajectoryDataset method) get_field_normalizers() (campd.data.normalization.normalization.DatasetNormalizer method) get_goal() (campd.data.context.TrajectoryContext method) get_hard_conditions() (campd.data.context.TrajectoryContext method) (campd.data.trajectory_sample.TrajectorySample method) get_item() (campd.data.context.TrajectoryContext method) get_limits() (campd.data.normalization.normalization.NormalizationCfg method) get_mask() (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) get_means() (campd.data.normalization.normalization.NormalizationCfg method) get_start() (campd.data.context.TrajectoryContext method) get_stds() (campd.data.normalization.normalization.NormalizationCfg method) get_torch_device() (in module campd.utils.torch) global_step (campd.training.base.Trainer attribute) goal (campd.data.context.TrajectoryContext attribute) group_norm_n_groups() (in module campd.architectures.layers.layers) H has_limits() (campd.data.normalization.normalization.NormalizationCfg method) has_means() (campd.data.normalization.normalization.NormalizationCfg method) has_nan() (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) (campd.data.trajectory_sample.TrajectorySample method) has_stds() (campd.data.normalization.normalization.NormalizationCfg method) hdf5_file (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) HDF5FieldCfg (class in campd.data.trajectory_dataset) hidden_dim (campd.architectures.layers.layers.MLP1DCfg attribute) I Identity (class in campd.data.normalization.normalization) impl_path() (in module campd.utils.registry) import_string() (in module campd.utils.registry) in_dim (campd.architectures.layers.layers.MLP1DCfg attribute) include_start_goal (campd.architectures.context.encoder.ContextEncoderCfg attribute) inference_scheduler_cls (campd.models.diffusion.model.SamplingCfg attribute) inference_scheduler_kwargs (campd.models.diffusion.model.SamplingCfg attribute) init (campd.utils.registry.Spec attribute) init_() (in module campd.architectures.layers.utils) is_batched (campd.data.context.TrajectoryContext attribute) (campd.data.embedded_context.EmbeddedContext attribute) (campd.data.trajectory_sample.TrajectorySample attribute) is_empty() (campd.data.normalization.normalization.NormalizationCfg method) is_normalized (campd.data.context.TrajectoryContext attribute) (campd.data.trajectory_sample.TrajectorySample attribute) K key_networks (campd.architectures.context.encoder.ContextEncoder attribute) (campd.architectures.context.encoder.ContextEncoderCfg attribute) KeyNetModule (class in campd.architectures.context.encoder) keys() (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) (campd.utils.torch.TensorArgs method) L last_batch (campd.training.base.Trainer attribute) layer_norm (campd.architectures.layers.layers.MLP1DCfg attribute) LayerNorm (class in campd.architectures.layers.layers) LimitsNormalizer (class in campd.data.normalization.normalization) load() (campd.data.normalization.normalization.DatasetNormalizer static method) load_only_context (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) load_params_from_yaml() (in module campd.utils.file) load_yaml() (in module campd.utils.file) log_every_n_steps (campd.training.base.CUDAGraphCfg attribute) loss_fn (campd.training.objectives.diffusion.DiffusionObjectiveCfg attribute) M main() (in module campd.runner) masks (campd.data.embedded_context.EmbeddedContext attribute) max_epochs (campd.training.base.TrainerCfg attribute) max_neg_value() (in module campd.architectures.layers.utils) maxs (campd.data.normalization.normalization.FieldLimit attribute) merge_dict_b_in_a() (in module campd.utils.config) metadata (campd.data.trajectory_sample.TrajectorySample attribute) mins (campd.data.normalization.normalization.FieldLimit attribute) MLP1D (class in campd.architectures.layers.layers) MLP1DCfg (class in campd.architectures.layers.layers) model (campd.training.base.Trainer attribute) model_config (campd.architectures.context.encoder.ContextEncoderCfg attribute) (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) (campd.architectures.layers.layers.MLP1DCfg attribute) (campd.data.normalization.normalization.FieldLimit attribute) (campd.data.normalization.normalization.NormalizationCfg attribute) (campd.data.trajectory_dataset.HDF5FieldCfg attribute) (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) (campd.models.diffusion.model.SamplingCfg attribute) (campd.training.base.CUDAGraphCfg attribute) (campd.training.base.TrainerCfg attribute) (campd.training.objectives.diffusion.DiffusionObjectiveCfg attribute) (campd.utils.registry.Spec attribute) (campd.utils.torch.TensorArgs attribute) model_post_init() (campd.data.trajectory_dataset.HDF5FieldCfg method) (campd.utils.torch.TensorArgs method) model_type (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) module campd.architectures.context.encoder campd.architectures.diffusion.base campd.architectures.diffusion.temporal_unet campd.architectures.layers.attention campd.architectures.layers.layers campd.architectures.layers.utils campd.data.context campd.data.embedded_context campd.data.normalization.base campd.data.normalization.normalization campd.data.trajectory_dataset campd.data.trajectory_sample campd.models.diffusion.model campd.models.helpers campd.runner campd.training.base campd.training.callbacks campd.training.losses campd.training.objectives.base campd.training.objectives.diffusion campd.training.summary campd.utils.config campd.utils.file campd.utils.random campd.utils.registry campd.utils.stats campd.utils.torch N n_diffusion_steps (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) n_layers (campd.architectures.layers.layers.MLP1DCfg attribute) n_support_points (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) (campd.models.diffusion.model.SamplingCfg attribute) (campd.training.base.CUDAGraphCfg attribute) network (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) noise_scheduler (campd.models.diffusion.model.ContextTrajectoryDiffusionModel attribute) normalization_config (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) NormalizationCfg (class in campd.data.normalization.normalization) normalize() (campd.data.context.TrajectoryContext method) (campd.data.normalization.base.Normalizer method) (campd.data.normalization.normalization.DatasetNormalizer method) (campd.data.normalization.normalization.GaussianNormalizer method) (campd.data.normalization.normalization.Identity method) (campd.data.normalization.normalization.LimitsNormalizer method) (campd.data.trajectory_sample.TrajectorySample method) Normalize() (in module campd.architectures.layers.attention) normalize_delta() (campd.data.normalization.base.Normalizer method) (campd.data.normalization.normalization.DatasetNormalizer method) (campd.data.normalization.normalization.GaussianNormalizer method) (campd.data.normalization.normalization.Identity method) (campd.data.normalization.normalization.LimitsNormalizer method) normalizer (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) Normalizer (class in campd.data.normalization.base) normalizer_class (campd.data.normalization.normalization.NormalizationCfg attribute) null() (campd.data.embedded_context.EmbeddedContext method) num_inference_steps (campd.models.diffusion.model.SamplingCfg attribute) O objective (campd.training.base.TrainerCfg attribute) on_batch_end() (campd.training.callbacks.Callback method) (campd.training.callbacks.CheckpointCallback method) (campd.training.callbacks.EMACallback method) (campd.training.callbacks.WandBCallback method) on_batch_start() (campd.training.callbacks.Callback method) on_epoch_end() (campd.training.callbacks.Callback method) (campd.training.callbacks.CheckpointCallback method) on_epoch_start() (campd.training.callbacks.Callback method) on_fit_start() (campd.training.callbacks.Callback method) on_summary_end() (campd.training.callbacks.Callback method) (campd.training.callbacks.WandBCallback method) on_train_end() (campd.training.callbacks.Callback method) (campd.training.callbacks.PrinterCallback method) on_train_start() (campd.training.callbacks.Callback method) (campd.training.callbacks.EMACallback method) (campd.training.callbacks.PrinterCallback method) on_validation_end() (campd.training.callbacks.Callback method) (campd.training.callbacks.CheckpointCallback method) (campd.training.callbacks.EarlyStoppingCallback method) (campd.training.callbacks.WandBCallback method) on_validation_start() (campd.training.callbacks.Callback method) optimizer (campd.training.base.TrainerCfg attribute) out_dim (campd.architectures.layers.layers.MLP1DCfg attribute) P PreNorm (class in campd.architectures.layers.layers) prepare_for_sampling() (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) PrinterCallback (class in campd.training.callbacks) prob_mask_like() (in module campd.architectures.layers.utils) process_imports() (in module campd.utils.config) propagate_attributes() (in module campd.utils.config) propagate_attributes_dict() (in module campd.utils.config) propagate_parent_attributes() (in module campd.utils.config) Q q_dim (campd.data.trajectory_dataset.HDF5FieldCfg attribute) R random_split() (campd.data.trajectory_dataset.TrajectoryDataset method) register() (campd.utils.registry.Registry method) registry (campd.utils.registry.Spec attribute) Registry (class in campd.utils.registry) registry_key_for() (in module campd.utils.registry) repeat_interleave() (campd.data.context.TrajectoryContext method) rescaled_phi (campd.models.diffusion.model.SamplingCfg attribute) Residual (class in campd.architectures.layers.layers) ResidualTemporalBlock (class in campd.architectures.layers.layers) results_dir (campd.training.base.Trainer property) (campd.training.base.TrainerCfg attribute) return_chain (campd.models.diffusion.model.SamplingCfg attribute) ReverseDiffusionNetwork (class in campd.architectures.diffusion.base) run() (campd.training.summary.Summary method) S SafeLimitsNormalizer (class in campd.data.normalization.normalization) sample() (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) SamplingCfg (class in campd.models.diffusion.model) save() (campd.data.normalization.normalization.DatasetNormalizer static method) save_checkpoint() (campd.training.callbacks.CheckpointCallback method) save_config() (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) save_field_config() (campd.data.trajectory_dataset.TrajectoryDataset method) save_params_to_yaml() (in module campd.utils.file) scheduler (campd.training.base.TrainerCfg attribute) scheduler_interval (campd.training.base.TrainerCfg attribute) serialize_device() (campd.utils.torch.TensorArgs method) serialize_dtype() (campd.utils.torch.TensorArgs method) set_limits() (campd.data.normalization.normalization.NormalizationCfg method) set_means() (campd.data.normalization.normalization.NormalizationCfg method) set_normalizer() (campd.data.trajectory_dataset.TrajectoryDataset method) set_stds() (campd.data.normalization.normalization.NormalizationCfg method) shared_context (campd.data.trajectory_sample.TrajectorySample attribute) should_run() (campd.training.summary.Summary method) SinusoidalPosEmb (class in campd.architectures.layers.layers) slice() (campd.data.context.TrajectoryContext method) (campd.data.trajectory_sample.TrajectorySample method) SpatialTransformer (class in campd.architectures.layers.attention) Spec (class in campd.utils.registry) spec_payload() (in module campd.utils.registry) start (campd.data.context.TrajectoryContext attribute) state_dim (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg attribute) (campd.training.base.CUDAGraphCfg attribute) step() (campd.training.objectives.base.TrainingObjective method) (campd.training.objectives.diffusion.DiffusionObjective method) stop_training (campd.training.base.Trainer attribute) storage_device (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) str_to_torch_dtype() (in module campd.utils.torch) summaries (campd.training.base.TrainerCfg attribute) summarize_1d() (in module campd.utils.stats) summarize_2d() (in module campd.utils.stats) summarize_all_stats() (in module campd.utils.stats) Summary (class in campd.training.summary) T TemporalUnet (class in campd.architectures.diffusion.temporal_unet) TemporalUnetCfg (class in campd.architectures.diffusion.temporal_unet) tensor_args (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) (campd.training.base.TrainerCfg attribute) TensorArgs (class in campd.utils.torch) time_emb_dim (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) TimeEncoder (class in campd.architectures.layers.layers) TimerCUDA (class in campd.utils.torch) timesteps (campd.models.diffusion.model.SamplingCfg attribute) to() (campd.data.context.TrajectoryContext method) (campd.data.embedded_context.EmbeddedContext method) (campd.data.normalization.normalization.DatasetNormalizer method) (campd.data.trajectory_sample.TrajectorySample method) (campd.models.diffusion.model.ContextTrajectoryDiffusionModel method) to_torch() (in module campd.utils.torch) to_yamlable() (in module campd.utils.file) torchjd_aggregator (campd.training.base.TrainerCfg attribute) train_dataloader (campd.training.base.Trainer attribute) Trainer (class in campd.training.base) TrainerCfg (class in campd.training.base) TrainingObjective (class in campd.training.objectives.base) trajectory (campd.data.trajectory_sample.TrajectorySample attribute) trajectory_field (campd.data.trajectory_dataset.HDF5FieldCfg attribute) trajectory_state (campd.data.trajectory_dataset.TrajectoryDatasetCfg attribute) TrajectoryContext (class in campd.data.context) TrajectoryDataset (class in campd.data.trajectory_dataset) TrajectoryDatasetCfg (class in campd.data.trajectory_dataset) TrajectorySample (class in campd.data.trajectory_sample) U unet_input_dim (campd.architectures.diffusion.temporal_unet.TemporalUnetCfg attribute) uniq() (in module campd.architectures.layers.utils) unnormalize() (campd.data.normalization.base.Normalizer method) (campd.data.normalization.normalization.DatasetNormalizer method) (campd.data.normalization.normalization.GaussianNormalizer method) (campd.data.normalization.normalization.Identity method) (campd.data.normalization.normalization.LimitsNormalizer method) update_average() (campd.training.callbacks.EMA method) update_model_average() (campd.training.callbacks.EMA method) Upsample1d (class in campd.architectures.layers.layers) use_amp (campd.training.base.TrainerCfg attribute) use_cuda_graph (campd.models.diffusion.model.SamplingCfg attribute) use_torchjd (campd.training.base.TrainerCfg attribute) V val_check_interval (campd.training.base.TrainerCfg attribute) val_dataloader (campd.training.base.Trainer attribute) validate_batch_size() (campd.data.context.TrajectoryContext method) validate_device() (campd.utils.torch.TensorArgs class method) validate_dtype() (campd.utils.torch.TensorArgs class method) validate_for_run() (campd.models.diffusion.model.ContextTrajectoryDiffusionModelCfg method) validate_trajectory_state() (campd.data.trajectory_dataset.TrajectoryDatasetCfg class method) W WandBCallback (class in campd.training.callbacks) warmup_iters (campd.training.base.CUDAGraphCfg attribute) WeightedL1 (class in campd.training.losses) WeightedL2 (class in campd.training.losses) WeightedLoss (class in campd.training.losses) which_has_nan() (campd.data.context.TrajectoryContext method) Z zero_module() (in module campd.architectures.layers.attention)