Index A | B | C | D | E | F | G | H | I | L | M | N | O | P | R | S | T | U | W A AimLogger (class in compressai_trainer.utils.catalyst.loggers) AllSuperlogger (class in compressai_trainer.utils.catalyst.loggers) B BaseRunner (class in compressai_trainer.runners) batch_meters (compressai_trainer.runners.BaseRunner attribute) best_metric_index() (in module compressai_trainer.utils.aim.query) branch_name() (in module compressai_trainer.utils.git) build_args() (in module compressai_trainer.run.plot_rd) C close_log() (compressai_trainer.utils.catalyst.loggers.AimLogger method) commit_count() (in module compressai_trainer.utils.git) commit_hash() (in module compressai_trainer.utils.git) commit_version() (in module compressai_trainer.utils.git) common_ancestor_commit_hash() (in module compressai_trainer.utils.git) common_ancestor_commit_version() (in module compressai_trainer.utils.git) compressai_results_dataframe() (in module compressai_trainer.utils.compressai.results) compressai_trainer.config module compressai_trainer.config.config module compressai_trainer.config.dataset module compressai_trainer.config.engine module compressai_trainer.config.env module compressai_trainer.config.load module compressai_trainer.config.outputs module compressai_trainer.plot module compressai_trainer.registry module compressai_trainer.registry.catalyst module compressai_trainer.registry.torch module compressai_trainer.run.eval_model module compressai_trainer.run.plot_rd module compressai_trainer.run.train module compressai_trainer.runners module compressai_trainer.typing module compressai_trainer.utils module compressai_trainer.utils.aim.query module compressai_trainer.utils.catalyst.loggers module compressai_trainer.utils.compressai.results module compressai_trainer.utils.git module compressai_trainer.utils.metrics module compressai_trainer.utils.pip module compressai_trainer.utils.system module compressai_trainer.utils.utils module compute_metrics() (in module compressai_trainer.utils.metrics) compute_padding() (in module compressai_trainer.utils.utils) ConfigStringFormatter (class in compressai_trainer.utils.utils) configure_conf() (in module compressai_trainer.config.config) configure_engine() (in module compressai_trainer.config.engine) create_callback() (in module compressai_trainer.config.engine) create_criterion() (in module compressai_trainer.config.config) create_data_transform() (in module compressai_trainer.config.dataset) create_data_transform_composition() (in module compressai_trainer.config.dataset) create_dataframe() (in module compressai_trainer.run.plot_rd) create_dataloader() (in module compressai_trainer.config.dataset) create_dataloaders() (in module compressai_trainer.config.config) create_dataset() (in module compressai_trainer.config.dataset) create_dataset_tuple() (in module compressai_trainer.config.dataset) create_logger() (in module compressai_trainer.config.engine) create_model() (in module compressai_trainer.config.config) create_module() (in module compressai_trainer.config.config) create_optimizer() (in module compressai_trainer.config.config) create_runner() (in module compressai_trainer.config.engine) create_scheduler() (in module compressai_trainer.config.config) criterion (compressai_trainer.runners.BaseRunner attribute) D dataset (compressai_trainer.config.dataset.DatasetTuple attribute) DatasetTuple (class in compressai_trainer.config.dataset) db() (in module compressai_trainer.utils.metrics) diff() (in module compressai_trainer.utils.git) DistributionSuperlogger (class in compressai_trainer.utils.catalyst.loggers) dl_to_ld() (in module compressai_trainer.utils.utils) E exclude (compressai_trainer.utils.catalyst.loggers.AimLogger attribute) F featuremap_image() (in module compressai_trainer.plot) featuremap_matplotlib() (in module compressai_trainer.plot) FigureSuperlogger (class in compressai_trainer.utils.catalyst.loggers) flatten_values() (in module compressai_trainer.utils.utils) format_dataframe() (in module compressai_trainer.utils.utils) freeze() (in module compressai_trainer.utils.pip) G get_checkpoint_path() (in module compressai_trainer.config.load) get_env() (in module compressai_trainer.config.env) get_field() (compressai_trainer.utils.utils.ConfigStringFormatter method) get_runs_dataframe() (in module compressai_trainer.utils.aim.query) GVAEImageCompressionRunner (class in compressai_trainer.runners) H handle_batch() (compressai_trainer.runners.GVAEImageCompressionRunner method) (compressai_trainer.runners.ImageCompressionRunner method) hostname() (in module compressai_trainer.utils.system) I ImageCompressionRunner (class in compressai_trainer.runners) ImageSuperlogger (class in compressai_trainer.utils.catalyst.loggers) L ld_to_dl() (in module compressai_trainer.utils.utils) list() (in module compressai_trainer.utils.pip) load_checkpoint() (in module compressai_trainer.config.load) load_config() (in module compressai_trainer.config.load) load_model() (in module compressai_trainer.config.load) loader (compressai_trainer.config.dataset.DatasetTuple attribute) log_artifact() (compressai_trainer.utils.catalyst.loggers.AimLogger method) log_distribution() (compressai_trainer.utils.catalyst.loggers.AimLogger method) (compressai_trainer.utils.catalyst.loggers.DistributionSuperlogger method) log_figure() (compressai_trainer.utils.catalyst.loggers.AimLogger method) (compressai_trainer.utils.catalyst.loggers.FigureSuperlogger method) log_hparams() (compressai_trainer.utils.catalyst.loggers.AimLogger method) log_image() (compressai_trainer.runners.BaseRunner method) (compressai_trainer.utils.catalyst.loggers.AimLogger method) (compressai_trainer.utils.catalyst.loggers.ImageSuperlogger method) log_metrics() (compressai_trainer.utils.catalyst.loggers.AimLogger method) logger (compressai_trainer.utils.catalyst.loggers.AimLogger property) loggers (compressai_trainer.utils.catalyst.loggers.DistributionSuperlogger attribute) (compressai_trainer.utils.catalyst.loggers.FigureSuperlogger attribute) (compressai_trainer.utils.catalyst.loggers.ImageSuperlogger attribute) M main() (in module compressai_trainer.run.eval_model) (in module compressai_trainer.run.plot_rd) (in module compressai_trainer.run.train) main_branch_name() (in module compressai_trainer.utils.git) metrics_at_index() (in module compressai_trainer.utils.aim.query) model (compressai_trainer.runners.BaseRunner attribute) model_module (compressai_trainer.runners.BaseRunner property) module compressai_trainer.config compressai_trainer.config.config compressai_trainer.config.dataset compressai_trainer.config.engine compressai_trainer.config.env compressai_trainer.config.load compressai_trainer.config.outputs compressai_trainer.plot compressai_trainer.registry compressai_trainer.registry.catalyst compressai_trainer.registry.torch compressai_trainer.run.eval_model compressai_trainer.run.plot_rd compressai_trainer.run.train compressai_trainer.runners compressai_trainer.typing compressai_trainer.utils compressai_trainer.utils.aim.query compressai_trainer.utils.catalyst.loggers compressai_trainer.utils.compressai.results compressai_trainer.utils.git compressai_trainer.utils.metrics compressai_trainer.utils.pip compressai_trainer.utils.system compressai_trainer.utils.utils msssim() (in module compressai_trainer.utils.metrics) msssim_db() (in module compressai_trainer.utils.metrics) N np_img_to_tensor() (in module compressai_trainer.utils.utils) num_parameters() (in module compressai_trainer.utils.utils) O on_epoch_end() (compressai_trainer.runners.BaseRunner method) on_epoch_start() (compressai_trainer.runners.BaseRunner method) on_experiment_end() (compressai_trainer.runners.BaseRunner method) on_experiment_start() (compressai_trainer.runners.BaseRunner method) on_loader_end() (compressai_trainer.runners.BaseRunner method) (compressai_trainer.runners.GVAEImageCompressionRunner method) (compressai_trainer.runners.ImageCompressionRunner method) on_loader_start() (compressai_trainer.runners.BaseRunner method) (compressai_trainer.runners.GVAEImageCompressionRunner method) (compressai_trainer.runners.ImageCompressionRunner method) optimizer (compressai_trainer.runners.BaseRunner attribute) P plot_dataframe() (in module compressai_trainer.run.plot_rd) plot_entropy_bottleneck_distributions() (in module compressai_trainer.plot) plot_pdf_signatures() (in module compressai_trainer.plot) plot_rd() (in module compressai_trainer.plot) predict_batch() (compressai_trainer.runners.GVAEImageCompressionRunner method) (compressai_trainer.runners.ImageCompressionRunner method) psnr() (in module compressai_trainer.utils.metrics) R register_callback() (in module compressai_trainer.registry.catalyst) register_criterion() (in module compressai_trainer.registry.torch) register_dataset() (in module compressai_trainer.registry.torch) register_model() (in module compressai_trainer.registry.torch) register_module() (in module compressai_trainer.registry.torch) register_optimizer() (in module compressai_trainer.registry.torch) register_runner() (in module compressai_trainer.registry.catalyst) register_scheduler() (in module compressai_trainer.registry.torch) run (compressai_trainer.utils.catalyst.loggers.AimLogger attribute) run_eval_model() (in module compressai_trainer.run.eval_model) run_hashes_by_query() (in module compressai_trainer.utils.aim.query) S setup() (in module compressai_trainer.run.eval_model) (in module compressai_trainer.run.train) state_dict_from_checkpoint() (in module compressai_trainer.config.load) T TCallback (in module compressai_trainer.typing) TCriterion (in module compressai_trainer.typing) TDataLoader (in module compressai_trainer.typing) TDataset (in module compressai_trainer.typing) tensor_to_np_img() (in module compressai_trainer.utils.utils) TModel (in module compressai_trainer.typing) TModule (in module compressai_trainer.typing) transform (compressai_trainer.config.dataset.DatasetTuple attribute) TRunner (in module compressai_trainer.typing) U username() (in module compressai_trainer.utils.system) W wrap() (in module compressai_trainer.run.plot_rd) write_config() (in module compressai_trainer.config.outputs) write_git_diff() (in module compressai_trainer.config.outputs) write_outputs() (in module compressai_trainer.config.outputs) write_pip_list() (in module compressai_trainer.config.outputs) write_pip_requirements() (in module compressai_trainer.config.outputs)
