curobo.opt.particle package¶
This module contains particle-based optimization solvers.
Submodules¶
- curobo.opt.particle.parallel_es module
ParallelESConfigParallelESConfig.learning_rateParallelESConfig.create_data_dictParallelESConfig.init_meanParallelESConfig.init_covParallelESConfig.base_actionParallelESConfig.step_size_meanParallelESConfig.step_size_covParallelESConfig.null_act_fracParallelESConfig.squash_fnParallelESConfig.cov_typeParallelESConfig.sample_paramsParallelESConfig.update_covParallelESConfig.random_meanParallelESConfig.betaParallelESConfig.alphaParallelESConfig.gammaParallelESConfig.kappaParallelESConfig.sample_per_problemParallelESConfig.sample_modeParallelESConfig.seedParallelESConfig.calculate_valueParallelESConfig.store_rolloutsParallelESConfig.d_actionParallelESConfig.action_lowsParallelESConfig.action_highsParallelESConfig.action_horizonParallelESConfig.horizonParallelESConfig.n_itersParallelESConfig.cold_start_n_itersParallelESConfig.rollout_fnParallelESConfig.tensor_argsParallelESConfig.use_cuda_graphParallelESConfig.store_debugParallelESConfig.debug_infoParallelESConfig.n_problemsParallelESConfig.num_particlesParallelESConfig.sync_cuda_timeParallelESConfig.use_coo_sparse
ParallelESParallelES._compute_meanParallelES._exp_util_from_costsParallelES._exp_utilParallelES._compute_mean_covarianceParallelES._update_distributionParallelES._calc_valParallelES._call_cuda_opt_itersParallelES._compute_covarianceParallelES._compute_total_costParallelES._get_action_seqParallelES._initialize_cuda_graphParallelES._optimizeParallelES._run_opt_itersParallelES._shiftParallelES._update_cov_scaleParallelES._update_problem_kernelParallelES.check_convergenceParallelES.create_data_dictParallelES.entropyParallelES.full_covParallelES.full_inv_covParallelES.full_scale_trilParallelES.generate_noiseParallelES.generate_rolloutsParallelES.get_all_rollout_instancesParallelES.get_nproblem_tensorParallelES.get_rolloutsParallelES.learning_rateParallelES.optimizeParallelES.resetParallelES.reset_covarianceParallelES.reset_cuda_graphParallelES.reset_distributionParallelES.reset_meanParallelES.reset_seedParallelES.reset_shapeParallelES.sample_actionsParallelES.squashed_meanParallelES.update_init_meanParallelES.update_nproblemsParallelES.update_num_particles_per_problemParallelES.update_paramsParallelES.update_samplesParallelES.update_seedParallelES.d_actionParallelES.action_lowsParallelES.action_highsParallelES.action_horizonParallelES.horizonParallelES.n_itersParallelES.cold_start_n_itersParallelES.rollout_fnParallelES.tensor_argsParallelES.use_cuda_graphParallelES.store_debugParallelES.debug_infoParallelES.n_problemsParallelES.num_particlesParallelES.sync_cuda_timeParallelES.use_coo_sparseParallelES.init_meanParallelES.init_covParallelES.base_actionParallelES.step_size_meanParallelES.step_size_covParallelES.null_act_fracParallelES.squash_fnParallelES.cov_typeParallelES.sample_paramsParallelES.update_covParallelES.random_meanParallelES.betaParallelES.alphaParallelES.gammaParallelES.kappaParallelES.sample_per_problemParallelES.sample_modeParallelES.seedParallelES.calculate_valueParallelES.store_rollouts
calc_expcompute_es_mean
- curobo.opt.particle.parallel_mppi module
BaseActionTypeCovTypeParallelMPPIConfigParallelMPPIConfig.init_meanParallelMPPIConfig.init_covParallelMPPIConfig.base_actionParallelMPPIConfig.step_size_meanParallelMPPIConfig.step_size_covParallelMPPIConfig.null_act_fracParallelMPPIConfig.squash_fnParallelMPPIConfig.cov_typeParallelMPPIConfig.sample_paramsParallelMPPIConfig.update_covParallelMPPIConfig.random_meanParallelMPPIConfig.betaParallelMPPIConfig.alphaParallelMPPIConfig.gammaParallelMPPIConfig.kappaParallelMPPIConfig.sample_per_problemParallelMPPIConfig.create_data_dictParallelMPPIConfig.sample_modeParallelMPPIConfig.seedParallelMPPIConfig.calculate_valueParallelMPPIConfig.store_rolloutsParallelMPPIConfig.d_actionParallelMPPIConfig.action_lowsParallelMPPIConfig.action_highsParallelMPPIConfig.action_horizonParallelMPPIConfig.horizonParallelMPPIConfig.n_itersParallelMPPIConfig.cold_start_n_itersParallelMPPIConfig.rollout_fnParallelMPPIConfig.tensor_argsParallelMPPIConfig.use_cuda_graphParallelMPPIConfig.store_debugParallelMPPIConfig.debug_infoParallelMPPIConfig.n_problemsParallelMPPIConfig.num_particlesParallelMPPIConfig.sync_cuda_timeParallelMPPIConfig.use_coo_sparse
ParallelMPPIParallelMPPI.get_rolloutsParallelMPPI.reset_distributionParallelMPPI._compute_total_costParallelMPPI._exp_utilParallelMPPI._exp_util_from_costsParallelMPPI._compute_meanParallelMPPI._compute_mean_covarianceParallelMPPI._compute_covarianceParallelMPPI._update_cov_scaleParallelMPPI._update_distributionParallelMPPI.sample_actionsParallelMPPI.update_seedParallelMPPI.update_init_meanParallelMPPI.reset_meanParallelMPPI.reset_covarianceParallelMPPI._get_action_seqParallelMPPI.generate_noiseParallelMPPI._calc_valParallelMPPI.resetParallelMPPI.squashed_meanParallelMPPI.full_covParallelMPPI.full_inv_covParallelMPPI.full_scale_trilParallelMPPI.entropyParallelMPPI.reset_seedParallelMPPI.update_samplesParallelMPPI.generate_rolloutsParallelMPPI._call_cuda_opt_itersParallelMPPI._initialize_cuda_graphParallelMPPI._optimizeParallelMPPI._run_opt_itersParallelMPPI._shiftParallelMPPI._update_problem_kernelParallelMPPI.check_convergenceParallelMPPI.create_data_dictParallelMPPI.get_all_rollout_instancesParallelMPPI.get_nproblem_tensorParallelMPPI.optimizeParallelMPPI.reset_cuda_graphParallelMPPI.reset_shapeParallelMPPI.update_nproblemsParallelMPPI.update_num_particles_per_problemParallelMPPI.update_paramsParallelMPPI.d_actionParallelMPPI.action_lowsParallelMPPI.action_highsParallelMPPI.action_horizonParallelMPPI.horizonParallelMPPI.n_itersParallelMPPI.cold_start_n_itersParallelMPPI.rollout_fnParallelMPPI.tensor_argsParallelMPPI.use_cuda_graphParallelMPPI.store_debugParallelMPPI.debug_infoParallelMPPI.n_problemsParallelMPPI.num_particlesParallelMPPI.sync_cuda_timeParallelMPPI.use_coo_sparseParallelMPPI.init_meanParallelMPPI.init_covParallelMPPI.base_actionParallelMPPI.step_size_meanParallelMPPI.step_size_covParallelMPPI.null_act_fracParallelMPPI.squash_fnParallelMPPI.cov_typeParallelMPPI.sample_paramsParallelMPPI.update_covParallelMPPI.random_meanParallelMPPI.betaParallelMPPI.alphaParallelMPPI.gammaParallelMPPI.kappaParallelMPPI.sample_per_problemParallelMPPI.sample_modeParallelMPPI.seedParallelMPPI.calculate_valueParallelMPPI.store_rollouts
jit_calculate_exp_utiljit_calculate_exp_util_from_costsjit_compute_total_costjit_diag_a_cov_updatejit_blend_covjit_blend_meanjit_mean_cov_diag_a
- curobo.opt.particle.particle_opt_base module
SampleModeParticleOptInfoParticleOptConfigParticleOptConfig.gammaParticleOptConfig.sample_modeParticleOptConfig.seedParticleOptConfig.calculate_valueParticleOptConfig.store_rolloutsParticleOptConfig.create_data_dictParticleOptConfig.d_actionParticleOptConfig.action_lowsParticleOptConfig.action_highsParticleOptConfig.action_horizonParticleOptConfig.horizonParticleOptConfig.n_itersParticleOptConfig.cold_start_n_itersParticleOptConfig.rollout_fnParticleOptConfig.tensor_argsParticleOptConfig.use_cuda_graphParticleOptConfig.store_debugParticleOptConfig.debug_infoParticleOptConfig.n_problemsParticleOptConfig.num_particlesParticleOptConfig.sync_cuda_timeParticleOptConfig.use_coo_sparse
ParticleOptBaseParticleOptBase._get_action_seqParticleOptBase.sample_actionsParticleOptBase.update_seedParticleOptBase._update_distributionParticleOptBase.resetParticleOptBase._calc_valParticleOptBase.check_convergenceParticleOptBase.generate_rolloutsParticleOptBase._optimizeParticleOptBase._initialize_cuda_graphParticleOptBase._call_cuda_opt_itersParticleOptBase._run_opt_itersParticleOptBase.update_nproblemsParticleOptBase._shiftParticleOptBase._update_problem_kernelParticleOptBase.create_data_dictParticleOptBase.get_all_rollout_instancesParticleOptBase.get_nproblem_tensorParticleOptBase.optimizeParticleOptBase.reset_cuda_graphParticleOptBase.reset_seedParticleOptBase.reset_shapeParticleOptBase.update_num_particles_per_problemParticleOptBase.update_paramsParticleOptBase.d_actionParticleOptBase.action_lowsParticleOptBase.action_highsParticleOptBase.action_horizonParticleOptBase.horizonParticleOptBase.n_itersParticleOptBase.cold_start_n_itersParticleOptBase.rollout_fnParticleOptBase.tensor_argsParticleOptBase.use_cuda_graphParticleOptBase.store_debugParticleOptBase.debug_infoParticleOptBase.n_problemsParticleOptBase.num_particlesParticleOptBase.sync_cuda_timeParticleOptBase.use_coo_sparseParticleOptBase.gammaParticleOptBase.sample_modeParticleOptBase.seedParticleOptBase.calculate_valueParticleOptBase.store_rollouts
- curobo.opt.particle.particle_opt_utils module