Index _ | A | D | E | F | G | H | K | L | M | P | R | S | W _ __call__() (gnn_tracking_hpo.slurmcontrol.SlurmControl method) (gnn_tracking_hpo.tune.Dispatcher method) __version__ (in module gnn_tracking_hpo) _default_base_path (in module gnn_tracking_hpo.util.paths) _get_actions() (gnn_tracking_hpo.slurmcontrol.SlurmControl method) _get_new_model() (gnn_tracking_hpo.trainable.ECTrainable method) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) _get_restored_model() (gnn_tracking_hpo.trainable.ECTrainable method) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) _handle_action() (gnn_tracking_hpo.slurmcontrol.SlurmControl method) _is_continued_run (gnn_tracking_hpo.trainable.ECTrainable property) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable property) _need_edge_loss (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable property) _parse_slurm_time_str() (in module gnn_tracking_hpo.slurmcontrol) _refresh() (gnn_tracking_hpo.slurmcontrol.SlurmControl method) _update_config_from_restored_tc() (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) _update_edge_loss_config_from_ec() (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) A add_common_options() (in module gnn_tracking_hpo.tune) add_cpu_option() (in module gnn_tracking_hpo.cli) add_ec_restore_options() (in module gnn_tracking_hpo.cli) add_enqueue_option() (in module gnn_tracking_hpo.cli) add_local_option() (in module gnn_tracking_hpo.cli) add_scripts_path() (in module gnn_tracking_hpo.util.paths) add_tc_restore_options() (in module gnn_tracking_hpo.cli) add_test_option() (in module gnn_tracking_hpo.cli) add_truth_cut_options() (in module gnn_tracking_hpo.cli) add_wandb_options() (in module gnn_tracking_hpo.cli) auto_suggest_if_not_fixed() (in module gnn_tracking_hpo.config) D dbscan_float (in module gnn_tracking_hpo.dbscan_gpu) DefaultTrainable (class in gnn_tracking_hpo.trainable) Dispatcher (class in gnn_tracking_hpo.tune) dispatcher_id (gnn_tracking_hpo.trainable.DefaultTrainable attribute) E ECTrainable (class in gnn_tracking_hpo.trainable) F find_checkpoint() (in module gnn_tracking_hpo.util.paths) find_result_dir() (in module gnn_tracking_hpo.util.paths) fixed_dbscan_scan() (in module gnn_tracking_hpo.cluster_scans) G gdf_float (in module gnn_tracking_hpo.dbscan_gpu) get_background_loss_function() (gnn_tracking_hpo.trainable.DefaultTrainable method) get_callbacks() (gnn_tracking_hpo.tune.Dispatcher method) get_checkpoint_config() (gnn_tracking_hpo.tune.Dispatcher method) get_cluster_functions() (gnn_tracking_hpo.trainable.DefaultTrainable method) (gnn_tracking_hpo.trainable.ECTrainable method) get_config() (in module gnn_tracking_hpo.util.paths) get_edge_loss_function() (gnn_tracking_hpo.trainable.DefaultTrainable method) get_graphs_separate() (in module gnn_tracking_hpo.load) get_graphs_split() (in module gnn_tracking_hpo.load) get_loaders() (gnn_tracking_hpo.trainable.DefaultTrainable method) (in module gnn_tracking_hpo.load) get_logger() (in module gnn_tracking_hpo.util.log) get_loss_functions() (gnn_tracking_hpo.trainable.DefaultTrainable method) (gnn_tracking_hpo.trainable.ECTrainable method) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) get_lr_scheduler() (gnn_tracking_hpo.trainable.DefaultTrainable method) get_metadata() (in module gnn_tracking_hpo.config) get_model() (gnn_tracking_hpo.trainable.DefaultTrainable method) (gnn_tracking_hpo.trainable.ECTrainable method) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) get_no_improvement_stopper() (gnn_tracking_hpo.tune.Dispatcher method) get_num_samples() (gnn_tracking_hpo.tune.Dispatcher method) get_optimizer() (gnn_tracking_hpo.trainable.DefaultTrainable method) get_optuna_sampler() (gnn_tracking_hpo.tune.Dispatcher method) get_optuna_search() (gnn_tracking_hpo.tune.Dispatcher method) get_points_to_evaluate() (in module gnn_tracking_hpo.config) get_potential_loss_function() (gnn_tracking_hpo.trainable.DefaultTrainable method) get_resources() (gnn_tracking_hpo.tune.Dispatcher method) get_run_config() (gnn_tracking_hpo.tune.Dispatcher method) get_scheduler() (gnn_tracking_hpo.tune.Dispatcher method) get_slurm_job_id() (in module gnn_tracking_hpo.slurmcontrol) get_slurm_remaining_minutes() (in module gnn_tracking_hpo.slurmcontrol) get_stoppers() (gnn_tracking_hpo.tune.Dispatcher method) get_timeout_stopper() (in module gnn_tracking_hpo.tune) get_trainer() (gnn_tracking_hpo.trainable.DefaultTrainable method) (gnn_tracking_hpo.trainable.ECTrainable method) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) get_tune_config() (gnn_tracking_hpo.tune.Dispatcher method) get_tuner() (gnn_tracking_hpo.tune.Dispatcher method) get_wandb_callbacks() (gnn_tracking_hpo.tune.Dispatcher method) gnn_tracking_hpo module gnn_tracking_hpo.cli module gnn_tracking_hpo.cluster_scans module gnn_tracking_hpo.config module gnn_tracking_hpo.dbscan_gpu module gnn_tracking_hpo.defaults module gnn_tracking_hpo.load module gnn_tracking_hpo.orchestrate module gnn_tracking_hpo.restore module gnn_tracking_hpo.slurmcontrol module gnn_tracking_hpo.trainable module gnn_tracking_hpo.tune module gnn_tracking_hpo.util module gnn_tracking_hpo.util.dict module gnn_tracking_hpo.util.log module gnn_tracking_hpo.util.paths module H have_internet() (in module gnn_tracking_hpo.orchestrate) hook_before_trainer_setup() (gnn_tracking_hpo.trainable.DefaultTrainable method) (gnn_tracking_hpo.trainable.PretrainedECTCNTrainable method) HPOTrainable (class in gnn_tracking_hpo.trainable) K KILL_NODE (gnn_tracking_hpo.slurmcontrol.SlurmControlAction attribute) kill_slurm_job() (in module gnn_tracking_hpo.slurmcontrol) L legacy_config_compatibility() (in module gnn_tracking_hpo.defaults) load_checkpoint() (gnn_tracking_hpo.trainable.DefaultTrainable method) LOG_DEFAULT_LEVEL (in module gnn_tracking_hpo.util.log) logger (in module gnn_tracking_hpo.util.log) M main() (in module gnn_tracking_hpo.tune) maybe_run_distributed() (in module gnn_tracking_hpo.orchestrate) maybe_run_wandb_offline() (in module gnn_tracking_hpo.orchestrate) module gnn_tracking_hpo gnn_tracking_hpo.cli gnn_tracking_hpo.cluster_scans gnn_tracking_hpo.config gnn_tracking_hpo.dbscan_gpu gnn_tracking_hpo.defaults gnn_tracking_hpo.load gnn_tracking_hpo.orchestrate gnn_tracking_hpo.restore gnn_tracking_hpo.slurmcontrol gnn_tracking_hpo.trainable gnn_tracking_hpo.tune gnn_tracking_hpo.util gnn_tracking_hpo.util.dict gnn_tracking_hpo.util.log gnn_tracking_hpo.util.paths P parse_slurm_time_str() (in module gnn_tracking_hpo.slurmcontrol) points_to_evaluate() (gnn_tracking_hpo.tune.Dispatcher method) pop() (in module gnn_tracking_hpo.util.dict) PretrainedECTCNTrainable (class in gnn_tracking_hpo.trainable) R read_json() (in module gnn_tracking_hpo.config) reduced_dbscan_scan() (in module gnn_tracking_hpo.cluster_scans) reinstate() (gnn_tracking_hpo.trainable.HPOTrainable class method) restore_model() (in module gnn_tracking_hpo.restore) retrieve_config_from_wandb() (in module gnn_tracking_hpo.config) S save_checkpoint() (gnn_tracking_hpo.trainable.DefaultTrainable method) setup() (gnn_tracking_hpo.trainable.DefaultTrainable method) simple_run_without_tune() (in module gnn_tracking_hpo.tune) SlurmControl (class in gnn_tracking_hpo.slurmcontrol) SlurmControlAction (class in gnn_tracking_hpo.slurmcontrol) step() (gnn_tracking_hpo.trainable.DefaultTrainable method) suggest_default_values() (in module gnn_tracking_hpo.defaults) W WAIT (gnn_tracking_hpo.slurmcontrol.SlurmControlAction attribute)