Ablator package

Subpackages

• Analysis package
  • Subpackages
    • Plot package
      • Submodules
      • Categorical plot module
      • Plot analysis module
      • Numerical plot module module
      • Plotting utilities module
      • Module contents
  • Submodules
  • Analysis module module
    • Analysis
      • Analysis.__init__()
  • Analysis results module
    • Results
      • Results.__init__()
      • Results.metric_names
      • Results.read_results()
    • process_row()
    • read_result()
  • Module contents
• Config package
  • Submodules
  • Base Configuration module
    • ConfigBase
      • ConfigBase.__init__()
      • ConfigBase.annotations
      • ConfigBase.assert_state()
      • ConfigBase.assert_unambigious()
      • ConfigBase.diff()
      • ConfigBase.diff_str()
      • ConfigBase.get_annot_type_with_dot_path()
      • ConfigBase.get_type_with_dot_path()
      • ConfigBase.get_val_with_dot_path()
      • ConfigBase.keys()
      • ConfigBase.load()
      • ConfigBase.make_dict()
      • ConfigBase.merge()
      • ConfigBase.to_dict()
      • ConfigBase.to_dot_path()
      • ConfigBase.to_str()
      • ConfigBase.to_yaml()
      • ConfigBase.uid
      • ConfigBase.write()
    • Missing
    • configclass()
  • Configuration Types module
    • Annotation
      • Annotation.collection
      • Annotation.optional
      • Annotation.state
      • Annotation.variable_type
    • Derived
    • Dict
    • Enum
    • List
    • Optional
    • Stateful
    • Stateless
    • Tuple
    • get_annotation_state()
    • parse_type_hint()
    • parse_value()
  • Configuration Utils module
    • dict_hash()
    • flatten_nested_dict()
  • Module contents
• Main package
  • Subpackages
    • Main model package
      • Submodules
      • Main Model module
      • Model Wrapper module
      • Module contents
  • Submodules
  • Model Configuration module
    • ModelConfig
      • ModelConfig.config_class
    • Optim
    • ParallelConfig
      • ParallelConfig.config_class
    • RunConfig
      • RunConfig.config_class
      • RunConfig.uid
    • SearchAlgo
    • SearchSpace
      • SearchSpace.__init__()
      • SearchSpace.config_class
    • SearchType
    • TrainConfig
      • TrainConfig.config_class
  • Multi-process Trainer module
    • ParallelTrainer
      • ParallelTrainer.__init__()
      • ParallelTrainer.evaluate()
      • ParallelTrainer.kill_idle()
      • ParallelTrainer.launch()
      • ParallelTrainer.sync_down()
      • ParallelTrainer.sync_up()
    • parse_metrics()
    • parse_rsync_paths()
    • train_main_remote()
  • Prototype Trainer module
    • ProtoTrainer
      • ProtoTrainer.__init__()
      • ProtoTrainer.evaluate()
      • ProtoTrainer.launch()
      • ProtoTrainer.pre_train_setup()
      • ProtoTrainer.smoke_test()
      • ProtoTrainer.sync()
  • Experiment and Optuna state module
    • Base
    • OptunaState
      • OptunaState.__init__()
      • OptunaState.sample_trial()
      • OptunaState.update_trial()
    • Trial
    • TrialState
      • TrialState.to_optuna_state()
    • ExperimentState
      • ExperimentState.__init__()
      • ExperimentState.n_trials_remaining
      • ExperimentState.pruned_errored_trials
      • ExperimentState.sample_trials()
      • ExperimentState.search_space_dot_path()
      • ExperimentState.tune_trial_str()
      • ExperimentState.update_trial_state()
    • augment_trial_kwargs()
    • parse_metrics()
    • sample_trial_params()
  • Module contents
• Modules package
  • Subpackages
    • Loggers package
      • Submodules
      • File Logger module
      • Main Logger module
      • Tensorboard Logger module
      • Module contents
    • Metrics package
      • Submodules
      • Main Metrics module
      • Metric Stores module
      • Module contents
    • Storage package
      • Submodules
      • GCloud Storage module
      • Remote Storage module
      • Module contents
  • Submodules
  • Optimizers module
    • AdamConfig
      • AdamConfig.config_class
      • AdamConfig.init_optimizer()
    • AdamWConfig
      • AdamWConfig.config_class
      • AdamWConfig.init_optimizer()
    • OptimizerArgs
      • OptimizerArgs.config_class
      • OptimizerArgs.init_optimizer()
    • OptimizerConfig
      • OptimizerConfig.__init__()
      • OptimizerConfig.config_class
      • OptimizerConfig.make_optimizer()
    • SGDConfig
      • SGDConfig.config_class
      • SGDConfig.init_optimizer()
    • get_optim_parameters()
    • get_parameter_names()
  • Schedulers module
    • OneCycleConfig
      • OneCycleConfig.config_class
      • OneCycleConfig.init_scheduler()
    • PlateuaConfig
      • PlateuaConfig.config_class
      • PlateuaConfig.init_scheduler()
    • SchedulerArgs
      • SchedulerArgs.config_class
      • SchedulerArgs.init_scheduler()
    • SchedulerConfig
      • SchedulerConfig.__init__()
      • SchedulerConfig.config_class
      • SchedulerConfig.make_scheduler()
    • StepLRConfig
      • StepLRConfig.config_class
      • StepLRConfig.init_scheduler()
  • Module contents
• Utils package
  • Submodules
  • Base Utils module
    • Dummy
      • Dummy.__init__()
    • apply_lambda_to_iter()
    • debugger_is_active()
    • get_gpu_max_mem()
    • get_gpu_mem()
    • get_latest_chkpts()
    • get_lr()
    • init_weights()
    • iter_to_device()
    • iter_to_numpy()
    • parse_device()
    • set_seed()
  • File Utils module
    • clean_checkpoints()
    • default_val_parser()
    • dict_to_json()
    • json_to_dict()
    • make_sub_dirs()
    • nested_set()
    • save_checkpoint()
  • Module contents

Module contents