Index

_

__init__() (ablator.analysis.main.Analysis method) (ablator.analysis.plot.cat_plot.Categorical method) (ablator.analysis.plot.cat_plot.ViolinPlot method) (ablator.analysis.plot.Plot method) (ablator.analysis.results.Results method) (ablator.config.main.ConfigBase method) (ablator.main.configs.SearchSpace method) (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) (ablator.main.mp.ParallelTrainer method) (ablator.main.proto.ProtoTrainer method) (ablator.main.state.ExperimentState method) (ablator.main.state.OptunaState method) (ablator.modules.loggers.file.FileLogger method) (ablator.modules.loggers.main.SummaryLogger method) (ablator.modules.loggers.tensor.TensorboardLogger method) (ablator.modules.metrics.main.TrainMetrics method) (ablator.modules.metrics.stores.ArrayStore method) (ablator.modules.metrics.stores.PredictionStore method) (ablator.modules.optimizer.OptimizerConfig method) (ablator.modules.scheduler.SchedulerConfig method) (ablator.modules.storage.cloud.GcpConfig method) (ablator.utils.base.Dummy method)

A

ablator module ablator.analysis module ablator.analysis.main module ablator.analysis.plot module ablator.analysis.plot.cat_plot module ablator.analysis.plot.main module ablator.analysis.plot.num_plot module ablator.analysis.plot.utils module ablator.analysis.results module ablator.config module ablator.config.main module ablator.config.types module ablator.config.utils module ablator.main module ablator.main.configs module ablator.main.model module ablator.main.model.main module ablator.main.model.wrapper module ablator.main.mp module ablator.main.proto module ablator.modules module ablator.modules.loggers module ablator.modules.loggers.file module

ablator.modules.loggers.main module ablator.modules.loggers.tensor module ablator.modules.metrics module ablator.modules.metrics.main module ablator.modules.metrics.stores module ablator.modules.optimizer module ablator.modules.scheduler module ablator.modules.storage module ablator.modules.storage.cloud module ablator.modules.storage.remote module ablator.utils module ablator.utils.base module ablator.utils.file module AdamConfig (class in ablator.modules.optimizer) AdamWConfig (class in ablator.modules.optimizer) add_image() (ablator.modules.loggers.tensor.TensorboardLogger method) add_scalar() (ablator.modules.loggers.tensor.TensorboardLogger method) add_scalars() (ablator.modules.loggers.tensor.TensorboardLogger method) add_table() (ablator.modules.loggers.tensor.TensorboardLogger method) add_text() (ablator.modules.loggers.tensor.TensorboardLogger method) Analysis (class in ablator.analysis.main) Annotation (class in ablator.config.types) annotations (ablator.config.main.ConfigBase property) append() (ablator.modules.metrics.stores.ArrayStore method) (ablator.modules.metrics.stores.MovingAverage method) (ablator.modules.metrics.stores.PredictionStore method) append_batch() (ablator.modules.metrics.main.TrainMetrics method) apply_lambda_to_iter() (in module ablator.utils.base) apply_loss() (ablator.main.model.wrapper.ModelWrapper method) ArrayStore (class in ablator.modules.metrics.stores) assert_state() (ablator.config.main.ConfigBase method) assert_unambigious() (ablator.config.main.ConfigBase method) augment_trial_kwargs() (in module ablator.main.state) aux_metrics() (ablator.main.model.wrapper.ModelWrapper method)

B

Base (class in ablator.main.state)

C

Categorical (class in ablator.analysis.plot.cat_plot) checkpoint() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) (ablator.modules.loggers.main.SummaryLogger method) CheckpointNotFoundError clean_checkpoints() (ablator.modules.loggers.main.SummaryLogger method) (in module ablator.utils.file) collection (ablator.config.types.Annotation attribute) config_class (ablator.main.configs.ModelConfig attribute) (ablator.main.configs.ParallelConfig attribute) (ablator.main.configs.RunConfig attribute) (ablator.main.configs.SearchSpace attribute) (ablator.main.configs.TrainConfig attribute) (ablator.modules.loggers.LoggerConfig attribute) (ablator.modules.optimizer.AdamConfig attribute) (ablator.modules.optimizer.AdamWConfig attribute) (ablator.modules.optimizer.OptimizerArgs attribute) (ablator.modules.optimizer.OptimizerConfig attribute) (ablator.modules.optimizer.SGDConfig attribute) (ablator.modules.scheduler.OneCycleConfig attribute) (ablator.modules.scheduler.PlateuaConfig attribute) (ablator.modules.scheduler.SchedulerArgs attribute) (ablator.modules.scheduler.SchedulerConfig attribute) (ablator.modules.scheduler.StepLRConfig attribute) (ablator.modules.storage.cloud.GcpConfig attribute) (ablator.modules.storage.remote.RemoteConfig attribute)

config_parser() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) ConfigBase (class in ablator.config.main) configclass() (in module ablator.config.main) create_model() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) create_optimizer() (ablator.main.model.wrapper.ModelWrapper method) create_scaler() (ablator.main.model.wrapper.ModelWrapper method) create_scheduler() (ablator.main.model.wrapper.ModelWrapper method) current_epoch (ablator.main.model.main.ModelBase property)

D

debugger_is_active() (in module ablator.utils.base) default_val_parser() (in module ablator.utils.file) Derived (class in ablator.config.types) Dict (class in ablator.config.types) dict_hash() (in module ablator.config.utils)

dict_to_json() (in module ablator.utils.file) diff() (ablator.config.main.ConfigBase method) diff_str() (ablator.config.main.ConfigBase method) Dummy (class in ablator.utils.base) DuplicateRunError

E

Enum (class in ablator.config.types) epoch_len (ablator.main.model.main.ModelBase property) error() (ablator.modules.loggers.file.FileLogger method) (ablator.modules.loggers.main.SummaryLogger method) eval() (ablator.main.model.wrapper.ModelWrapper method) eval_itr (ablator.main.model.main.ModelBase property) evaluate() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) (ablator.main.mp.ParallelTrainer method) (ablator.main.proto.ProtoTrainer method) (ablator.modules.metrics.main.TrainMetrics method) (ablator.modules.metrics.stores.PredictionStore method)

evaluation_functions() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) EvaluationError ExperimentState (class in ablator.main.state)

F

FileLogger (class in ablator.modules.loggers.file)

flatten_nested_dict() (in module ablator.config.utils)

G

GcpConfig (class in ablator.modules.storage.cloud) get() (ablator.modules.metrics.stores.ArrayStore method) get_annot_type_with_dot_path() (ablator.config.main.ConfigBase method) get_annotation_state() (in module ablator.config.types) get_gpu_max_mem() (in module ablator.utils.base) get_gpu_mem() (in module ablator.utils.base)

get_latest_chkpts() (in module ablator.utils.base) get_lr() (in module ablator.utils.base) get_optim_parameters() (in module ablator.modules.optimizer) get_parameter_names() (in module ablator.modules.optimizer) get_type_with_dot_path() (ablator.config.main.ConfigBase method) get_val_with_dot_path() (ablator.config.main.ConfigBase method)

I

info() (ablator.modules.loggers.file.FileLogger method) (ablator.modules.loggers.main.SummaryLogger method) init_optimizer() (ablator.modules.optimizer.AdamConfig method) (ablator.modules.optimizer.AdamWConfig method) (ablator.modules.optimizer.OptimizerArgs method) (ablator.modules.optimizer.SGDConfig method)

init_scheduler() (ablator.modules.scheduler.OneCycleConfig method) (ablator.modules.scheduler.PlateuaConfig method) (ablator.modules.scheduler.SchedulerArgs method) (ablator.modules.scheduler.StepLRConfig method) init_weights() (in module ablator.utils.base) iter_to_device() (in module ablator.utils.base) iter_to_numpy() (in module ablator.utils.base)

J

json_to_dict() (in module ablator.utils.file)

K

keys() (ablator.config.main.ConfigBase method)

kill_idle() (ablator.main.mp.ParallelTrainer method)

L

launch() (ablator.main.mp.ParallelTrainer method) (ablator.main.proto.ProtoTrainer method) LinearPlot (class in ablator.analysis.plot.num_plot) List (class in ablator.config.types) list_bucket() (ablator.modules.storage.cloud.GcpConfig method) load() (ablator.config.main.ConfigBase class method) load_checkpoint() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method)

log() (ablator.main.model.wrapper.ModelWrapper method) log_itr (ablator.main.model.main.ModelBase property) log_step() (ablator.main.model.wrapper.ModelWrapper method) LoggerBase (class in ablator.modules.loggers) LoggerConfig (class in ablator.modules.loggers) LogStepError LossDivergedError

M

make_dataloader_test() (ablator.main.model.wrapper.ModelWrapper method) make_dataloader_train() (ablator.main.model.wrapper.ModelWrapper method) make_dataloader_val() (ablator.main.model.wrapper.ModelWrapper method) make_dataloaders() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) make_dict() (ablator.config.main.ConfigBase method) make_optimizer() (ablator.modules.optimizer.OptimizerConfig method) make_scheduler() (ablator.modules.scheduler.SchedulerConfig method) make_sub_dirs() (in module ablator.utils.file) merge() (ablator.config.main.ConfigBase method) metric_names (ablator.analysis.results.Results property) Missing (class in ablator.config.main) mock_train() (ablator.main.model.wrapper.ModelWrapper method) model_step() (ablator.main.model.wrapper.ModelWrapper method) ModelBase (class in ablator.main.model.main) ModelConfig (class in ablator.main.configs) ModelWrapper (class in ablator.main.model.wrapper) module ablator ablator.analysis ablator.analysis.main ablator.analysis.plot ablator.analysis.plot.cat_plot ablator.analysis.plot.main ablator.analysis.plot.num_plot ablator.analysis.plot.utils ablator.analysis.results ablator.config ablator.config.main ablator.config.types ablator.config.utils ablator.main ablator.main.configs ablator.main.model ablator.main.model.main ablator.main.model.wrapper ablator.main.mp ablator.main.proto ablator.modules ablator.modules.loggers ablator.modules.loggers.file ablator.modules.loggers.main ablator.modules.loggers.tensor ablator.modules.metrics ablator.modules.metrics.main ablator.modules.metrics.stores ablator.modules.optimizer ablator.modules.scheduler ablator.modules.storage ablator.modules.storage.cloud ablator.modules.storage.remote ablator.utils ablator.utils.base ablator.utils.file

MovingAverage (class in ablator.modules.metrics.stores)

N

n_trials_remaining (ablator.main.state.ExperimentState property)

nested_set() (in module ablator.utils.file) Numerical (class in ablator.analysis.plot.num_plot)

O

OneCycleConfig (class in ablator.modules.scheduler) Optim (class in ablator.main.configs) OptimizerArgs (class in ablator.modules.optimizer)

OptimizerConfig (class in ablator.modules.optimizer) optional (ablator.config.types.Annotation attribute) Optional (class in ablator.config.types) OptunaState (class in ablator.main.state)

P

ParallelConfig (class in ablator.main.configs) ParallelTrainer (class in ablator.main.mp) parse_device() (in module ablator.utils.base) parse_metrics() (in module ablator.main.mp) (in module ablator.main.state) parse_name_remap() (in module ablator.analysis.plot.utils) parse_rsync_paths() (in module ablator.main.mp) parse_type_hint() (in module ablator.config.types)

parse_value() (in module ablator.config.types) PlateuaConfig (class in ablator.modules.scheduler) Plot (class in ablator.analysis.plot) PlotAnalysis (class in ablator.analysis.plot.main) pre_train_setup() (ablator.main.proto.ProtoTrainer method) PredictionStore (class in ablator.modules.metrics.stores) process_row() (in module ablator.analysis.results) ProtoTrainer (class in ablator.main.proto) pruned_errored_trials (ablator.main.state.ExperimentState property)

R

read_result() (in module ablator.analysis.results) read_results() (ablator.analysis.results.Results class method) RemoteConfig (class in ablator.modules.storage.remote) reset() (ablator.modules.metrics.main.TrainMetrics method) (ablator.modules.metrics.stores.ArrayStore method) (ablator.modules.metrics.stores.PredictionStore method) reset_optimizer_scheduler() (ablator.main.model.wrapper.ModelWrapper method)

Results (class in ablator.analysis.results) rsync_down() (ablator.modules.storage.cloud.GcpConfig method) (ablator.modules.storage.remote.RemoteConfig method) rsync_down_node() (ablator.modules.storage.cloud.GcpConfig method) rsync_up() (ablator.modules.storage.cloud.GcpConfig method) (ablator.modules.storage.remote.RemoteConfig method) run_cmd_wait() (in module ablator.modules.storage.remote) RunConfig (class in ablator.main.configs)

S

sample_trial() (ablator.main.state.OptunaState method) sample_trial_params() (in module ablator.main.state) sample_trials() (ablator.main.state.ExperimentState method) save_checkpoint() (in module ablator.utils.file) save_dict() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method) SchedulerArgs (class in ablator.modules.scheduler) SchedulerConfig (class in ablator.modules.scheduler) search_space_dot_path() (ablator.main.state.ExperimentState static method) SearchAlgo (class in ablator.main.configs) SearchSpace (class in ablator.main.configs) SearchType (class in ablator.main.configs) set_path() (ablator.modules.loggers.file.FileLogger method)

set_prefix() (ablator.modules.loggers.file.FileLogger method) set_seed() (in module ablator.utils.base) SGDConfig (class in ablator.modules.optimizer) smoke_test() (ablator.main.proto.ProtoTrainer method) state (ablator.config.types.Annotation attribute) Stateful (class in ablator.config.types) Stateless (class in ablator.config.types) status_message() (ablator.main.model.wrapper.ModelWrapper method) StepLRConfig (class in ablator.modules.scheduler) SummaryLogger (class in ablator.modules.loggers.main) sync() (ablator.main.proto.ProtoTrainer method) sync_down() (ablator.main.mp.ParallelTrainer method) sync_up() (ablator.main.mp.ParallelTrainer method)

T

TensorboardLogger (class in ablator.modules.loggers.tensor) to_device() (ablator.main.model.wrapper.ModelWrapper method) to_dict() (ablator.config.main.ConfigBase method) (ablator.modules.metrics.main.TrainMetrics method) to_dot_path() (ablator.config.main.ConfigBase method) to_optuna_state() (ablator.main.state.TrialState method) to_str() (ablator.config.main.ConfigBase method) to_yaml() (ablator.config.main.ConfigBase method) total_steps (ablator.main.model.wrapper.ModelWrapper property) train() (ablator.main.model.main.ModelBase method) (ablator.main.model.wrapper.ModelWrapper method)

train_loop() (ablator.main.model.wrapper.ModelWrapper method) train_main_remote() (in module ablator.main.mp) train_stats (ablator.main.model.main.ModelBase property) train_step() (ablator.main.model.wrapper.ModelWrapper method) TrainConfig (class in ablator.main.configs) TrainMetrics (class in ablator.modules.metrics.main) TrainPlateauError Trial (class in ablator.main.state) TrialState (class in ablator.main.state) tune_trial_str() (ablator.main.state.ExperimentState static method) Tuple (class in ablator.config.types)

U

uid (ablator.config.main.ConfigBase property) (ablator.main.configs.RunConfig property) (ablator.main.model.main.ModelBase property) update() (ablator.modules.loggers.main.SummaryLogger method)

update_ma_metrics() (ablator.modules.metrics.main.TrainMetrics method) update_static_metrics() (ablator.modules.metrics.main.TrainMetrics method) update_status() (ablator.main.model.wrapper.ModelWrapper method) update_trial() (ablator.main.state.OptunaState method) update_trial_state() (ablator.main.state.ExperimentState method)

V

validation_loop() (ablator.main.model.wrapper.ModelWrapper method)

variable_type (ablator.config.types.Annotation attribute) ViolinPlot (class in ablator.analysis.plot.cat_plot)

W

warn() (ablator.modules.loggers.file.FileLogger method) (ablator.modules.loggers.main.SummaryLogger method)

write() (ablator.config.main.ConfigBase method) write_config() (ablator.modules.loggers.tensor.TensorboardLogger method)