Index

A

accuracy (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule attribute) agent (common.optim.ne.fit.evolve parameter) (common.optim.ne.fit.fit parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.env_done_reset parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.run_post_eval parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.run_pre_eval parameter) (common.optim.ne.utils.initialize.initialize_agents parameter) agent_total_num_steps (common.optim.ne.utils.wandb.gather parameter) agents (common.optim.ne.space.base.BaseSpace.evaluate parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.evaluate parameter)

agents_batch (common.optim.ne.utils.compute.compute_generation_results parameter) (common.optim.ne.utils.evolve.evaluate_on_cpu parameter) (common.optim.ne.utils.evolve.evaluate_on_gpu parameter) (common.optim.ne.utils.evolve.mutate parameter) (common.optim.ne.utils.exchange.exchange_agents parameter) (common.optim.ne.utils.readwrite.save_state parameter) agents_per_task (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.utils.initialize.initialize_common_variables parameter)

B

BaseAgent (class in common.optim.ne.agent.base) BaseAgentConfig (class in common.optim.ne.agent.base) BaseClassificationLitModule (class in common.optim.dl.litmodule.classification.base) BaseClassificationLitModuleConfig (class in common.optim.dl.litmodule.classification.base) BaseDataModule (class in common.optim.dl.datamodule.base) BaseDataModuleConfig (class in common.optim.dl.datamodule.base) BaseHydraConfig (class in common.config) BaseLitModule (class in common.optim.dl.litmodule.base) BaseLitModuleConfig (class in common.optim.dl.litmodule.base) BaseReinforcementSpace (class in common.optim.ne.space.reinforcement)

BaseSpace (class in common.optim.ne.space.base) BaseSpaceConfig (class in common.optim.ne.space.base) BaseSubtaskConfig (class in common.config) BaseTaskRunner (class in common.runner) batch (common.optim.dl.litmodule.base.BaseLitModule.test_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.training_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.validation_step parameter) batch_idx (common.optim.dl.litmodule.base.BaseLitModule.optimizer_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.test_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.training_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.validation_step parameter)

C

can_connect_to_internet() (in module common.utils.misc) checkpoint (common.optim.dl.litmodule.base.BaseLitModule.on_load_checkpoint parameter) (common.optim.dl.litmodule.base.BaseLitModule.on_save_checkpoint parameter) ckpt_path (common.infer.lightning.config.LightningInferenceSubtaskConfig parameter) (common.optim.dl.config.DeepLearningSubtaskConfig parameter) clb (common.store.store_wandb_logger_configs parameter) (common.utils.misc.get_path parameter) cls (common.infer.lightning.runner.LightningInferTaskRunner.store_configs parameter) (common.runner.BaseTaskRunner.handle_configs parameter) (common.runner.BaseTaskRunner.store_configs parameter) collate_fn (common.optim.dl.datamodule.base.BaseDataModule attribute) common module common.config module common.infer module common.infer.config module common.infer.lightning module common.infer.lightning.config module common.infer.lightning.infer module common.infer.lightning.runner module common.infer.runner module common.optim module common.optim.config module common.optim.dl module common.optim.dl.config module common.optim.dl.datamodule module common.optim.dl.datamodule.base module common.optim.dl.litmodule module common.optim.dl.litmodule.base module common.optim.dl.litmodule.classification module common.optim.dl.litmodule.classification.base module common.optim.dl.litmodule.nnmodule module common.optim.dl.litmodule.nnmodule.mlp module common.optim.dl.litmodule.store module common.optim.dl.runner module common.optim.dl.store module common.optim.dl.train module common.optim.dl.utils module common.optim.dl.utils.lightning module common.optim.ne module common.optim.ne.agent module common.optim.ne.agent.base module common.optim.ne.config module common.optim.ne.fit module common.optim.ne.net module common.optim.ne.net.cpu module common.optim.ne.net.cpu.static module common.optim.ne.net.cpu.static.rnnfc module common.optim.ne.runner module common.optim.ne.space module common.optim.ne.space.base module

common.optim.ne.space.reinforcement module common.optim.ne.utils module common.optim.ne.utils.compute module common.optim.ne.utils.evolve module common.optim.ne.utils.exchange module common.optim.ne.utils.initialize module common.optim.ne.utils.readwrite module common.optim.ne.utils.type module common.optim.ne.utils.validate module common.optim.ne.utils.wandb module common.optim.runner module common.optim.store module common.optim.utils module common.optim.utils.hydra module common.runner module common.serve module common.store module common.test module common.utils module common.utils.beartype module common.utils.hydra_zen module common.utils.misc module common.utils.mpi4py module common.utils.runner module common.utils.torch module common.utils.wandb module compile (common.optim.dl.config.DeepLearningSubtaskConfig parameter) compute_generation_results() (in module common.optim.ne.utils.compute) compute_save_points() (in module common.optim.ne.utils.compute) compute_start_time_and_seeds() (in module common.optim.ne.utils.compute) compute_total_num_env_steps_and_process_fitnesses() (in module common.optim.ne.utils.compute) config (common.optim.dl.datamodule.base.BaseDataModule attribute) (common.optim.dl.datamodule.base.BaseDataModule parameter) (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule attribute) (common.optim.ne.agent.base.BaseAgent attribute) (common.optim.ne.agent.base.BaseAgent parameter) (common.optim.ne.fit.evolve parameter) (common.optim.ne.fit.fit parameter) (common.optim.ne.fit.test parameter) (common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFC parameter) (common.optim.ne.runner.NeuroevolutionTaskRunner.validate_subtask_config parameter) (common.optim.ne.space.base.BaseSpace parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace parameter) configure_optimizers() (common.optim.dl.litmodule.base.BaseLitModule method) continual_fitness (common.optim.ne.agent.base.BaseAgent attribute) CPUStaticRNNFC (class in common.optim.ne.net.cpu.static.rnnfc) CPUStaticRNNFCConfig (class in common.optim.ne.net.cpu.static.rnnfc) curr_episode_num_steps (common.optim.ne.agent.base.BaseAgent attribute) curr_episode_score (common.optim.ne.agent.base.BaseAgent attribute) curr_eval_num_steps (common.optim.ne.agent.base.BaseAgent attribute) curr_eval_score (common.optim.ne.agent.base.BaseAgent attribute) curr_gen (common.optim.ne.space.base.BaseSpace.evaluate parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.env_done_reset parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.evaluate parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.run_post_eval parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.run_pre_eval parameter) (common.optim.ne.utils.compute.compute_start_time_and_seeds parameter) (common.optim.ne.utils.compute.compute_total_num_env_steps_and_process_fitnesses parameter) (common.optim.ne.utils.evolve.evaluate_on_cpu parameter) (common.optim.ne.utils.evolve.evaluate_on_gpu parameter) (common.optim.ne.utils.readwrite.save_state parameter) (common.optim.ne.utils.wandb.gather parameter)

D

data (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.step parameter) data_dir (common.config.BaseSubtaskConfig parameter) (common.optim.dl.datamodule.base.BaseDataModuleConfig parameter) dataloader_idx (common.optim.dl.litmodule.base.BaseLitModule.test_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.training_step parameter) (common.optim.dl.litmodule.base.BaseLitModule.validation_step parameter) datamodule (common.optim.dl.utils.lightning.find_good_per_device_batch_size parameter) (common.optim.dl.utils.lightning.find_good_per_device_num_workers parameter) (common.optim.dl.utils.lightning.set_batch_size_and_num_workers parameter) dataset (common.optim.dl.datamodule.base.BaseDataModule.x_dataloader parameter) Datasets (class in common.optim.dl.datamodule.base)

datasets (common.optim.dl.datamodule.base.BaseDataModule attribute) DeepLearningSubtaskConfig (class in common.optim.dl.config) DeepLearningTaskConfig (class in common.optim.dl.config) DeepLearningTaskRunner (class in common.optim.dl.runner) destructure() (in module common.utils.hydra_zen) device (common.config.BaseSubtaskConfig parameter) (common.optim.dl.datamodule.base.BaseDataModuleConfig parameter) (common.optim.dl.utils.lightning.find_good_per_device_batch_size parameter) (common.optim.dl.utils.lightning.instantiate_trainer parameter) (common.optim.dl.utils.lightning.set_batch_size_and_num_workers parameter) device_ids (common.optim.dl.utils.lightning.find_good_per_device_batch_size parameter) drop_last (common.optim.dl.datamodule.base.BaseDataModuleConfig parameter)

E

element (common.utils.beartype.equal parameter) elements (common.utils.beartype.one_of parameter) env (common.optim.ne.space.reinforcement.BaseReinforcementSpace parameter) env_done_reset() (common.optim.ne.space.reinforcement.BaseReinforcementSpace method) env_transfer (common.optim.ne.agent.base.BaseAgentConfig parameter) (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) epoch (common.optim.dl.litmodule.base.BaseLitModule.optimizer_step parameter) equal() (in module common.utils.beartype) eval_num_steps (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.space.base.BaseSpaceConfig parameter)

evaluate() (common.optim.ne.space.base.BaseSpace method) (common.optim.ne.space.reinforcement.BaseReinforcementSpace method) evaluate_on_cpu() (in module common.optim.ne.utils.evolve) evaluate_on_gpu() (in module common.optim.ne.utils.evolve) evaluates_on_gpu (common.optim.ne.space.base.BaseSpace parameter) evolve() (in module common.optim.ne.fit) exchange_agents() (in module common.optim.ne.utils.exchange) exchange_and_mutate_info (common.optim.ne.utils.exchange.update_exchange_and_mutate_info parameter) exchange_and_mutate_info_batch (common.optim.ne.utils.evolve.mutate parameter) (common.optim.ne.utils.exchange.exchange_agents parameter)

F

find_existing_save_points() (in module common.optim.ne.utils.readwrite) find_good_per_device_batch_size() (in module common.optim.dl.utils.lightning) find_good_per_device_num_workers() (in module common.optim.dl.utils.lightning) fit() (in module common.optim.ne.fit) fit_transfer (common.optim.ne.agent.base.BaseAgentConfig parameter) (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter)

fitnesses_and_num_env_steps_batch (common.optim.ne.utils.compute.compute_generation_results parameter) fixed_per_device_batch_size (common.optim.dl.datamodule.base.BaseDataModuleConfig parameter) fixed_per_device_num_workers (common.optim.dl.datamodule.base.BaseDataModuleConfig parameter) forward() (common.optim.dl.litmodule.nnmodule.mlp.MLP method) (common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFC method)

G

gather() (in module common.optim.ne.utils.wandb) ge() (in module common.utils.beartype) generation_results (common.optim.ne.utils.compute.compute_generation_results parameter) (common.optim.ne.utils.compute.compute_start_time_and_seeds parameter) (common.optim.ne.utils.compute.compute_total_num_env_steps_and_process_fitnesses parameter) (common.optim.ne.utils.exchange.update_exchange_and_mutate_info parameter) (common.optim.ne.utils.readwrite.save_state parameter) generation_results_batch (common.optim.ne.utils.compute.compute_generation_results parameter)

get_launcher_config() (in module common.optim.utils.hydra) get_mpi_variables() (in module common.utils.mpi4py) get_path() (in module common.utils.misc) gt() (in module common.utils.beartype) GymAgent (class in projects.ne_control_score.agent) GymAgentConfig (class in projects.ne_control_score.agent) GymReinforcementSpace (class in projects.ne_control_score.space) GymReinforcementSpaceConfig (class in projects.ne_control_score.space)

H

handle_configs() (common.runner.BaseTaskRunner class method)

hidden_size (common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFCConfig parameter) hydra_config (common.runner.BaseTaskRunner attribute)

I

InferenceSubtaskConfig (class in common.infer.config) InferTaskRunner (class in common.infer.runner) initialize_agents() (in module common.optim.ne.utils.initialize) initialize_common_variables() (in module common.optim.ne.utils.initialize) initialize_eval_attributes() (common.optim.ne.agent.base.BaseAgent method)

initialize_gpu_comm() (in module common.optim.ne.utils.initialize) InitOptimParamsCheckpointConnector (class in common.optim.dl.utils.lightning) input_size (common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFCConfig parameter) instantiate_trainer() (in module common.optim.dl.utils.lightning) is_other_role_in_other_pop (common.optim.ne.agent.base.BaseAgent attribute) ith_gpu_comm (common.optim.ne.utils.evolve.evaluate_on_gpu parameter)

L

le() (in module common.utils.beartype) len_agents_batch (common.optim.ne.utils.initialize.initialize_agents parameter) (common.optim.ne.utils.readwrite.load_state parameter) LightningInferenceSubtaskConfig (class in common.infer.lightning.config) LightningInferenceTaskConfig (class in common.infer.lightning.config) LightningInferTaskRunner (class in common.infer.lightning.runner) litmodule (common.optim.dl.utils.lightning.find_good_per_device_batch_size parameter) (common.optim.dl.utils.lightning.set_batch_size_and_num_workers parameter) load_state() (in module common.optim.ne.utils.readwrite) load_state_dict() (common.optim.dl.datamodule.base.BaseDataModule method)

logged_score (common.optim.ne.utils.wandb.gather parameter) logger (common.optim.ne.fit.evolve parameter) (common.optim.ne.fit.fit parameter) (common.optim.ne.utils.wandb.setup_wandb parameter) logger_partial (common.optim.dl.utils.lightning.instantiate_trainer parameter) logging (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.config.NeuroevolutionSubtaskTestConfig parameter) (common.optim.ne.space.base.BaseSpaceConfig parameter) login_wandb() (in module common.utils.wandb) logits (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.save_wandb_data parameter) lt() (in module common.utils.beartype)

M

max_num_data_passes (common.optim.dl.utils.lightning.find_good_per_device_num_workers parameter) max_per_device_batch_size (common.optim.dl.datamodule.base.BaseDataModuleConfig parameter) mem_transfer (common.optim.ne.agent.base.BaseAgentConfig parameter) (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) MLP (class in common.optim.dl.litmodule.nnmodule.mlp) MLPConfig (class in common.optim.dl.litmodule.nnmodule.mlp) MNISTClassificationLitModule (class in projects.classify_mnist.litmodule) MNISTDataModule (class in projects.classify_mnist.datamodule) MNISTDataModuleConfig (class in projects.classify_mnist.datamodule) module common common.config common.infer common.infer.config common.infer.lightning common.infer.lightning.config common.infer.lightning.infer common.infer.lightning.runner common.infer.runner common.optim common.optim.config common.optim.dl common.optim.dl.config common.optim.dl.datamodule common.optim.dl.datamodule.base common.optim.dl.litmodule common.optim.dl.litmodule.base common.optim.dl.litmodule.classification common.optim.dl.litmodule.classification.base common.optim.dl.litmodule.nnmodule common.optim.dl.litmodule.nnmodule.mlp common.optim.dl.litmodule.store common.optim.dl.runner common.optim.dl.store common.optim.dl.train common.optim.dl.utils common.optim.dl.utils.lightning common.optim.ne common.optim.ne.agent common.optim.ne.agent.base common.optim.ne.config common.optim.ne.fit common.optim.ne.net common.optim.ne.net.cpu common.optim.ne.net.cpu.static common.optim.ne.net.cpu.static.rnnfc common.optim.ne.runner common.optim.ne.space common.optim.ne.space.base common.optim.ne.space.reinforcement common.optim.ne.utils common.optim.ne.utils.compute common.optim.ne.utils.evolve common.optim.ne.utils.exchange common.optim.ne.utils.initialize common.optim.ne.utils.readwrite common.optim.ne.utils.type common.optim.ne.utils.validate common.optim.ne.utils.wandb common.optim.runner common.optim.store common.optim.utils common.optim.utils.hydra common.runner common.serve common.store common.test common.utils common.utils.beartype common.utils.hydra_zen common.utils.misc common.utils.mpi4py common.utils.runner common.utils.torch common.utils.wandb projects projects.classify_mnist projects.classify_mnist.datamodule projects.classify_mnist.datamodule_test projects.classify_mnist.litmodule projects.ne_control_score projects.ne_control_score.agent projects.ne_control_score.space

mutate() (common.optim.ne.agent.base.BaseAgent method) (in module common.optim.ne.utils.evolve) (projects.ne_control_score.agent.GymAgent method)

N

NeuroevolutionSubtaskConfig (class in common.optim.ne.config) NeuroevolutionSubtaskTestConfig (class in common.optim.ne.config) NeuroevolutionTaskConfig (class in common.optim.ne.config) NeuroevolutionTaskRunner (class in common.optim.ne.runner) not_empty() (in module common.utils.beartype) num_classes (common.optim.dl.litmodule.classification.base.BaseClassificationLitModuleConfig parameter) num_pops (common.optim.ne.space.base.BaseSpace parameter) (common.optim.ne.utils.compute.compute_generation_results parameter) (common.optim.ne.utils.compute.compute_start_time_and_seeds parameter) (common.optim.ne.utils.evolve.mutate parameter) (common.optim.ne.utils.exchange.exchange_agents parameter) (common.optim.ne.utils.exchange.update_exchange_and_mutate_info parameter) (common.optim.ne.utils.initialize.initialize_agents parameter) (common.optim.ne.utils.initialize.initialize_common_variables parameter)

num_tests (common.optim.ne.config.NeuroevolutionSubtaskTestConfig parameter)

O

on_load_checkpoint() (common.optim.dl.litmodule.base.BaseLitModule method) on_save_checkpoint() (common.optim.dl.litmodule.base.BaseLitModule method) on_validation_end() (common.optim.dl.litmodule.base.BaseLitModule method) on_validation_epoch_end() (common.optim.dl.litmodule.base.BaseLitModule method) one_of() (in module common.utils.beartype) OptimizationSubtaskConfig (class in common.optim.config) optimizer (common.optim.dl.litmodule.base.BaseLitModule.optimizer_step parameter) optimizer_closure (common.optim.dl.litmodule.base.BaseLitModule.optimizer_step parameter) optimizer_step() (common.optim.dl.litmodule.base.BaseLitModule method) OptimTaskRunner (class in common.optim.runner) out (common.optim.ne.space.reinforcement.BaseReinforcementSpace.env_done_reset parameter) (common.optim.ne.space.reinforcement.BaseReinforcementSpace.run_post_eval parameter)

output_dir (common.config.BaseSubtaskConfig parameter) (common.optim.dl.utils.lightning.find_good_per_device_batch_size parameter) (common.optim.dl.utils.lightning.instantiate_trainer parameter) (common.optim.dl.utils.lightning.set_batch_size_and_num_workers parameter) (common.optim.ne.utils.compute.compute_save_points parameter) (common.optim.ne.utils.readwrite.find_existing_save_points parameter) (common.optim.ne.utils.readwrite.load_state parameter) (common.optim.ne.utils.readwrite.save_state parameter) (common.optim.ne.utils.wandb.setup_wandb parameter) output_size (common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFCConfig parameter)

P

per_device_batch_size (common.optim.dl.datamodule.base.BaseDataModule attribute) (common.optim.dl.utils.lightning.find_good_per_device_num_workers parameter) per_device_num_workers (common.optim.dl.datamodule.base.BaseDataModule attribute) pin_memory (common.optim.dl.datamodule.base.BaseDataModule attribute) pop_idx (common.optim.ne.agent.base.BaseAgent parameter) pop_merge (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.utils.compute.compute_start_time_and_seeds parameter) (common.optim.ne.utils.compute.compute_total_num_env_steps_and_process_fitnesses parameter) (common.optim.ne.utils.initialize.initialize_agents parameter) (common.optim.ne.utils.validate.validate_space parameter) pop_size (common.optim.ne.utils.compute.compute_start_time_and_seeds parameter) (common.optim.ne.utils.exchange.exchange_agents parameter) (common.optim.ne.utils.exchange.update_exchange_and_mutate_info parameter) pops_are_merged (common.optim.ne.agent.base.BaseAgent parameter) predict_dataloader() (common.optim.dl.datamodule.base.BaseDataModule method) prepare_data() (projects.classify_mnist.datamodule.MNISTDataModule method)

prev_num_gens (common.optim.ne.utils.readwrite.load_state parameter) projects module projects.classify_mnist module projects.classify_mnist.datamodule module projects.classify_mnist.datamodule_test module projects.classify_mnist.litmodule module projects.ne_control_score module projects.ne_control_score.agent module projects.ne_control_score.space module

R

reset() (common.optim.ne.agent.base.BaseAgent method) (common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFC method) (projects.ne_control_score.agent.GymAgent method) restore_optimizers() (common.optim.dl.utils.lightning.InitOptimParamsCheckpointConnector method) role (common.optim.ne.agent.base.BaseAgent attribute) run_post_eval() (common.optim.ne.space.reinforcement.BaseReinforcementSpace method)

run_pre_eval() (common.optim.ne.space.reinforcement.BaseReinforcementSpace method) run_subtask() (common.infer.lightning.runner.LightningInferTaskRunner class method) (common.optim.dl.runner.DeepLearningTaskRunner class method) (common.optim.ne.runner.NeuroevolutionTaskRunner class method) (common.runner.BaseTaskRunner static method) RunningStandardization (class in common.utils.torch)

S

save_every_n_train_steps (common.optim.dl.config.DeepLearningSubtaskConfig parameter) (common.optim.dl.utils.lightning.instantiate_trainer parameter) save_first_gen (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.utils.compute.compute_save_points parameter) save_interval (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.utils.compute.compute_save_points parameter) save_state() (in module common.optim.ne.utils.readwrite) save_wandb_data() (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule method) saved_env (common.optim.ne.agent.base.BaseAgent attribute) saved_env_out (common.optim.ne.agent.base.BaseAgent attribute) seed (common.utils.misc.seed_all parameter) seed_all() (in module common.utils.misc) seeds (common.optim.ne.utils.exchange.update_exchange_and_mutate_info parameter) set_batch_size_and_num_workers() (in module common.optim.dl.utils.lightning) setup() (projects.classify_mnist.datamodule.MNISTDataModule method) setup_wandb() (in module common.optim.ne.utils.wandb) shuffle (common.optim.dl.datamodule.base.BaseDataModule.x_dataloader parameter) space (common.optim.ne.fit.evolve parameter) (common.optim.ne.fit.fit parameter) (common.optim.ne.fit.test parameter) (common.optim.ne.utils.evolve.evaluate_on_cpu parameter) (common.optim.ne.utils.evolve.evaluate_on_gpu parameter) (common.optim.ne.utils.validate.validate_space parameter)

stage (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.save_wandb_data parameter) (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.step parameter) (projects.classify_mnist.datamodule.MNISTDataModule.setup parameter) start_time (common.optim.ne.utils.compute.compute_total_num_env_steps_and_process_fitnesses parameter) state_dict (common.optim.dl.datamodule.base.BaseDataModule.load_state_dict parameter) state_dict() (common.optim.dl.datamodule.base.BaseDataModule method) step() (common.optim.dl.litmodule.base.BaseLitModule method) (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule method) store (common.infer.lightning.runner.LightningInferTaskRunner.store_configs parameter) (common.optim.dl.runner.DeepLearningTaskRunner.store_configs parameter) (common.optim.dl.store.store_basic_trainer_config parameter) (common.optim.ne.runner.NeuroevolutionTaskRunner.store_configs parameter) (common.optim.runner.OptimTaskRunner.store_configs parameter) (common.optim.store.store_launcher_configs parameter) (common.runner.BaseTaskRunner.store_configs parameter) (common.store.store_wandb_logger_configs parameter) store_basic_trainer_config() (in module common.optim.dl.store) store_configs() (common.infer.lightning.runner.LightningInferTaskRunner class method) (common.optim.dl.runner.DeepLearningTaskRunner class method) (common.optim.ne.runner.NeuroevolutionTaskRunner class method) (common.optim.runner.OptimTaskRunner class method) (common.runner.BaseTaskRunner class method) store_launcher_configs() (in module common.optim.store) store_wandb_logger_configs() (in module common.store)

T

terminate_wandb() (in module common.optim.ne.utils.wandb) test() (in module common.optim.ne.fit) test_dataloader() (common.optim.dl.datamodule.base.BaseDataModule method) test_num_steps (common.optim.ne.config.NeuroevolutionSubtaskTestConfig parameter) test_step() (common.optim.dl.litmodule.base.BaseLitModule method) total_num_env_steps (common.optim.ne.utils.compute.compute_total_num_env_steps_and_process_fitnesses parameter) (common.optim.ne.utils.readwrite.save_state parameter) total_num_gens (common.optim.ne.config.NeuroevolutionSubtaskConfig parameter) (common.optim.ne.utils.compute.compute_save_points parameter)

total_num_steps (common.optim.ne.agent.base.BaseAgent attribute) train() (in module common.optim.dl.train) train_dataloader() (common.optim.dl.datamodule.base.BaseDataModule method) trainer (common.optim.dl.utils.lightning.set_batch_size_and_num_workers parameter) trainer_partial (common.optim.dl.utils.lightning.instantiate_trainer parameter) training_step() (common.optim.dl.litmodule.base.BaseLitModule method) transfer (common.optim.ne.utils.evolve.evaluate_on_gpu parameter)

U

update_exchange_and_mutate_info() (in module common.optim.ne.utils.exchange)

V

val (common.utils.beartype.ge parameter) (common.utils.beartype.gt parameter) (common.utils.beartype.le parameter) (common.utils.beartype.lt parameter)

val_dataloader() (common.optim.dl.datamodule.base.BaseDataModule method) validate_space() (in module common.optim.ne.utils.validate) validate_subtask_config() (common.optim.ne.runner.NeuroevolutionTaskRunner static method) validation_step() (common.optim.dl.litmodule.base.BaseLitModule method)

W

wandb_columns (common.optim.dl.litmodule.classification.base.BaseClassificationLitModuleConfig parameter) wandb_media_x (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule property) (projects.classify_mnist.litmodule.MNISTClassificationLitModule property)

wandb_table (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule attribute) wandb_train_log_interval (common.optim.dl.litmodule.base.BaseLitModuleConfig parameter)

X

x (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.save_wandb_data parameter)

x_dataloader() (common.optim.dl.datamodule.base.BaseDataModule method) x_size (common.utils.torch.RunningStandardization parameter)

Y

y (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.save_wandb_data parameter)

y_hat (common.optim.dl.litmodule.classification.base.BaseClassificationLitModule.save_wandb_data parameter)