compressai_trainer.typing

Types for type annotations.

compressai_trainer.typing.TCallback

alias of catalyst.core.callback.Callback

compressai_trainer.typing.TCriterion

alias of torch.nn.modules.module.Module

compressai_trainer.typing.TDataLoader

alias of torch.utils.data.dataloader.DataLoader

compressai_trainer.typing.TDataset

alias of torch.utils.data.dataset.Dataset

compressai_trainer.typing.TModel

alias of torch.nn.modules.module.Module

compressai_trainer.typing.TModule

alias of torch.nn.modules.module.Module

compressai_trainer.typing.TRunner

alias of catalyst.runners.runner.Runner