SYMBOL INDEX (344 symbols across 80 files)

FILE: bin/predict.py
  function main (line 6) | def main():

FILE: bin/train.py
  function main (line 6) | def main():

FILE: cifar_pipeline/configs/base.py
  function get_dataset (line 18) | def get_dataset(path, transforms, train, use_mixup):
  class ConfigCIFARBase (line 30) | class ConfigCIFARBase(ConfigBase):
    method __init__ (line 31) | def __init__(self, model, model_save_path, num_workers=8, batch_size=1...

FILE: cifar_pipeline/configs/fixup/base.py
  function get_dataset (line 21) | def get_dataset(path, transforms, train, use_mixup):
  class ConfigCIFARBase (line 33) | class ConfigCIFARBase(ConfigBase):
    method __init__ (line 34) | def __init__(self, model, model_save_path, num_workers=8, batch_size=1...

FILE: cifar_pipeline/configs/fixup/resnet110_bn.py
  class Config (line 10) | class Config(ConfigCIFARBase):
    method __init__ (line 11) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/resnet110_fixup.py
  class Config (line 10) | class Config(ConfigCIFARBase):
    method __init__ (line 11) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/resnet110_fixup_0_0_1.py
  class Config (line 10) | class Config(ConfigCIFARBase):
    method __init__ (line 11) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/resnet110_fixup_0_1.py
  class Config (line 10) | class Config(ConfigCIFARBase):
    method __init__ (line 11) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/resnet110_fixup_mixup.py
  class Config (line 10) | class Config(ConfigCIFARBase):
    method __init__ (line 11) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/base.py
  class ConfigWideResNetBase (line 12) | class ConfigWideResNetBase(ConfigCIFARBase):
    method __init__ (line 16) | def __init__(self, num_layers, fixup_coeff=1, normalization_type=BATCH...

FILE: cifar_pipeline/configs/fixup/wideresnet/batch_norm/10000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/batch_norm/1000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/batch_norm/100_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/batch_norm/10_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup/10000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup/1000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup/100_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup/10_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0/10000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0/1000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0/100_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0/10_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_0_1/10000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_0_1/1000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_0_1/100_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_0_1/10_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_1/10000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_1/1000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_1/100_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_0_1/10_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_10/10000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_10/1000_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_10/100_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/fixup/wideresnet/fixup_10/10_layers.py
  class Config (line 4) | class Config(ConfigWideResNetBase):
    method __init__ (line 5) | def __init__(self):

FILE: cifar_pipeline/configs/simple_cnn.py
  function get_model (line 20) | def get_model():
  class Config (line 39) | class Config(ConfigCIFARBase):
    method __init__ (line 40) | def __init__(self):

FILE: cifar_pipeline/dataset.py
  class CIFARDataset (line 5) | class CIFARDataset(data.Dataset):
    method __init__ (line 6) | def __init__(self, path, download=True, train=True):
    method get_image (line 9) | def get_image(self, item):
    method get_class (line 12) | def get_class(self, item):
    method __len__ (line 15) | def __len__(self):
    method __getitem__ (line 18) | def __getitem__(self, item):
  class CIFARImagesDataset (line 22) | class CIFARImagesDataset(CIFARDataset):
    method __getitem__ (line 23) | def __getitem__(self, item):
  class CIFARTargetsDataset (line 27) | class CIFARTargetsDataset(CIFARDataset):
    method __getitem__ (line 28) | def __getitem__(self, item):

FILE: cifar_pipeline/resnet_cifar.py
  function _weights_init (line 8) | def _weights_init(m):
  class LambdaLayer (line 13) | class LambdaLayer(nn.Module):
    method __init__ (line 14) | def __init__(self, lambd):
    method forward (line 18) | def forward(self, x):
  class BasicBlock (line 22) | class BasicBlock(nn.Module):
    method __init__ (line 26) | def __init__(self, in_planes, planes, stride=1, use_fixup=False, fixup...
    method forward (line 50) | def forward(self, x):
  class ResNet (line 62) | class ResNet(nn.Module):
    method __init__ (line 63) | def __init__(self, block, num_blocks, num_classes=10, use_fixup=False,...
    method _make_layer (line 90) | def _make_layer(self, block, planes, num_blocks, stride, use_fixup, fi...
    method forward (line 99) | def forward(self, x):
  function resnet110 (line 110) | def resnet110(use_fixup=False, fixup_coeff=1):

FILE: imagenet_pipeline/configs/base.py
  function get_dataset (line 21) | def get_dataset(path, transforms, use_mixup):
  class ConfigImageNetBase (line 33) | class ConfigImageNetBase(ConfigBase):
    method __init__ (line 34) | def __init__(self, model, model_save_path, num_workers=16, batch_size=...

FILE: imagenet_pipeline/configs/resnet101_fixup.py
  class Config (line 11) | class Config(ConfigImageNetBase):
    method __init__ (line 12) | def __init__(self, model_save_path=MODEL_SAVE_PATH):

FILE: imagenet_pipeline/configs/resnet101_fixup_128.py
  class Config (line 11) | class Config(ConfigImageNetBase):
    method __init__ (line 12) | def __init__(self, model_save_path=MODEL_SAVE_PATH):

FILE: imagenet_pipeline/configs/resnet50.py
  class Config (line 10) | class Config(ConfigImageNetBase):
    method __init__ (line 11) | def __init__(self, model_save_path=MODEL_SAVE_PATH):

FILE: imagenet_pipeline/configs/resnet50_fixup.py
  class Config (line 11) | class Config(ConfigImageNetBase):
    method __init__ (line 12) | def __init__(self, model_save_path=MODEL_SAVE_PATH):

FILE: imagenet_pipeline/configs/resnet50_fixup_128.py
  class Config (line 11) | class Config(ConfigImageNetBase):
    method __init__ (line 12) | def __init__(self, model_save_path=MODEL_SAVE_PATH):

FILE: imagenet_pipeline/dataset.py
  class ImageNetDataset (line 14) | class ImageNetDataset(data.Dataset):
    method __init__ (line 15) | def __init__(self, path):
    method get_image (line 30) | def get_image(self, item):
    method get_class (line 35) | def get_class(self, item):
    method __len__ (line 41) | def __len__(self):
    method __getitem__ (line 44) | def __getitem__(self, item):
  class ImageNetImagesDataset (line 48) | class ImageNetImagesDataset(ImageNetDataset):
    method __getitem__ (line 49) | def __getitem__(self, item):
  class ImageNetTargetsDataset (line 53) | class ImageNetTargetsDataset(ImageNetDataset):
    method __getitem__ (line 54) | def __getitem__(self, item):

FILE: mnist_pipeline/configs/base.py
  function get_dataset (line 22) | def get_dataset(mode, transforms):
  class ConfigMNISTBase (line 33) | class ConfigMNISTBase(ConfigBase):
    method __init__ (line 34) | def __init__(self, model, model_save_path, num_workers=4, batch_size=1...
  class PredictConfigMNISTBase (line 61) | class PredictConfigMNISTBase(PredictConfigBase):
    method __init__ (line 62) | def __init__(self, model, model_save_path, num_workers=4, batch_size=1...

FILE: mnist_pipeline/configs/resnet18.py
  class Config (line 8) | class Config(ConfigMNISTBase):
    method __init__ (line 9) | def __init__(self, model_save_path="models/resnet18"):

FILE: mnist_pipeline/configs/simple_cnn.py
  function get_model (line 11) | def get_model():
  class Config (line 30) | class Config(ConfigMNISTBase):
    method __init__ (line 31) | def __init__(self, model_save_path=MODEL_SAVE_PATH):
  class PredictConfig (line 35) | class PredictConfig(PredictConfigMNISTBase):
    method __init__ (line 36) | def __init__(self, model_save_path=MODEL_SAVE_PATH):

FILE: mnist_pipeline/dataset.py
  class MNISTDataset (line 11) | class MNISTDataset(data.Dataset):
    method __init__ (line 15) | def __init__(self, path, mode, val_ratio):
    method __len__ (line 29) | def __len__(self):
    method __getitem__ (line 32) | def __getitem__(self, item):
  class MNISTImagesDataset (line 40) | class MNISTImagesDataset(MNISTDataset):
    method __init__ (line 41) | def __init__(self, path, mode, val_ratio):
    method __getitem__ (line 44) | def __getitem__(self, item):
  class MNISTTargetsDataset (line 49) | class MNISTTargetsDataset(MNISTDataset):
    method __init__ (line 50) | def __init__(self, path, mode, val_ratio):
    method __getitem__ (line 53) | def __getitem__(self, item):

FILE: mnist_pipeline/tests/test_dataset.py
  class TestMNISTDataset (line 9) | class TestMNISTDataset:
    method setup (line 10) | def setup(self):
    method test_train_dataset (line 13) | def test_train_dataset(self):
    method test_val_dataset (line 24) | def test_val_dataset(self):
    method test_images_dataset (line 38) | def test_images_dataset(self):
    method test_targets_dataset (line 47) | def test_targets_dataset(self):

FILE: mnist_pipeline/tests/test_train.py
  class TestMNISTTrain (line 10) | class TestMNISTTrain:
    method test_mnist_train (line 11) | def test_mnist_train(self):

FILE: pipeline/config_base.py
  class ConfigBase (line 11) | class ConfigBase:
    method __init__ (line 12) | def __init__(
  class PredictConfigBase (line 62) | class PredictConfigBase:
    method __init__ (line 63) | def __init__(

FILE: pipeline/core.py
  class PipelineError (line 1) | class PipelineError(Exception):

FILE: pipeline/datasets/base.py
  class EmptyDataset (line 7) | class EmptyDataset(data.Dataset):
    method __len__ (line 8) | def __len__(self):
    method __getitem__ (line 11) | def __getitem__(self, item: int):
  class DatasetComposer (line 15) | class DatasetComposer(data.Dataset):
    method __init__ (line 16) | def __init__(self, datasets: Sequence):
    method __len__ (line 22) | def __len__(self):
    method __getitem__ (line 25) | def __getitem__(self, item: int):
  class OneHotTargetsDataset (line 29) | class OneHotTargetsDataset(data.Dataset):
    method __init__ (line 30) | def __init__(self, targets: Sequence, class_count: int):
    method __len__ (line 34) | def __len__(self):
    method __getitem__ (line 37) | def __getitem__(self, item: int):
  class MultiLabelTargetsDataset (line 44) | class MultiLabelTargetsDataset(data.Dataset):
    method __init__ (line 45) | def __init__(self, targets: Sequence, class_count: int):
    method __len__ (line 49) | def __len__(self):
    method __getitem__ (line 52) | def __getitem__(self, item: int):
  class DatasetWithPostprocessingFunc (line 62) | class DatasetWithPostprocessingFunc(data.Dataset):
    method __init__ (line 63) | def __init__(self, dataset, postprocessing_func):
    method __len__ (line 67) | def __len__(self):
    method __getitem__ (line 70) | def __getitem__(self, item):

FILE: pipeline/datasets/mixup.py
  class MixUpDatasetWrapper (line 6) | class MixUpDatasetWrapper(data.Dataset):
    method __init__ (line 7) | def __init__(self, dataset, alpha=1):
    method __len__ (line 12) | def __len__(self):
    method __getitem__ (line 15) | def __getitem__(self, item):

FILE: pipeline/logger.py
  function setup_logger (line 9) | def setup_logger(out_file=None, stderr=True, stderr_level=logging.INFO, ...

FILE: pipeline/losses/vector_cross_entropy.py
  class VectorCrossEntropy (line 5) | class VectorCrossEntropy(nn.Module):
    method __init__ (line 6) | def __init__(self):
    method forward (line 10) | def forward(self, input, target):

FILE: pipeline/metrics/accuracy.py
  class MetricsCalculatorAccuracy (line 9) | class MetricsCalculatorAccuracy(MetricsCalculatorBase):
    method __init__ (line 10) | def __init__(self, border=0.5):
    method zero_cache (line 15) | def zero_cache(self):
    method add (line 19) | def add(self, y_predicted, y_true):
    method calculate (line 23) | def calculate(self):

FILE: pipeline/metrics/base.py
  class MetricsCalculatorBase (line 4) | class MetricsCalculatorBase(abc.ABC):
    method zero_cache (line 6) | def zero_cache(self):
    method add (line 10) | def add(self, y_predicted, y_true):
    method calculate (line 14) | def calculate(self):
  class MetricsCalculatorEmpty (line 18) | class MetricsCalculatorEmpty(MetricsCalculatorBase):
    method zero_cache (line 19) | def zero_cache(self):
    method add (line 22) | def add(self, y_predicted, y_true):
    method calculate (line 25) | def calculate(self):

FILE: pipeline/models/base.py
  class Flatten (line 4) | class Flatten(nn.Module):
    method forward (line 5) | def forward(self, x):

FILE: pipeline/models/image_models/encoders/resnet.py
  class ResnetModelFeatureExtractorBase (line 5) | class ResnetModelFeatureExtractorBase(nn.Module):
    method __init__ (line 6) | def __init__(self, model, input_channels):
    method forward (line 23) | def forward(self, input):
  class Resnet18FeatureExtractor (line 27) | class Resnet18FeatureExtractor(ResnetModelFeatureExtractorBase):
    method __init__ (line 30) | def __init__(self, pretrained=True, input_channels=3):
  class Resnet34FeatureExtractor (line 37) | class Resnet34FeatureExtractor(ResnetModelFeatureExtractorBase):
    method __init__ (line 40) | def __init__(self, pretrained=True, input_channels=3):
  class Resnet50FeatureExtractor (line 47) | class Resnet50FeatureExtractor(ResnetModelFeatureExtractorBase):
    method __init__ (line 50) | def __init__(self, pretrained=True, input_channels=3):
  class Resnet101FeatureExtractor (line 57) | class Resnet101FeatureExtractor(ResnetModelFeatureExtractorBase):
    method __init__ (line 60) | def __init__(self, pretrained=True, input_channels=3):
  class Resnet152FeatureExtractor (line 67) | class Resnet152FeatureExtractor(ResnetModelFeatureExtractorBase):
    method __init__ (line 70) | def __init__(self, pretrained=True, input_channels=3):

FILE: pipeline/models/image_models/resnet_fixup.py
  class Bottleneck (line 6) | class Bottleneck(nn.Module):
    method __init__ (line 10) | def __init__(self, inplanes, planes, stride=1, downsample=None, fixup_...
    method forward (line 34) | def forward(self, x):
  class ResNet (line 54) | class ResNet(nn.Module):
    method __init__ (line 56) | def __init__(self, block, layers, num_classes=1000, input_channels=3):
    method _make_layer (line 79) | def _make_layer(self, block, planes, blocks, fixup_l, stride=1):
    method forward (line 93) | def forward(self, x):
  function resnet50 (line 110) | def resnet50(**kwargs):
  function resnet101 (line 115) | def resnet101(**kwargs):
  function resnet152 (line 120) | def resnet152(**kwargs):

FILE: pipeline/models/image_models/wide_resnet.py
  class BasicBlock (line 15) | class BasicBlock(nn.Module):
    method __init__ (line 16) | def __init__(self, in_planes, out_planes, stride, dropout=0.0):
    method forward (line 30) | def forward(self, x):
  class NetworkBlock (line 42) | class NetworkBlock(nn.Module):
    method __init__ (line 43) | def __init__(self, nb_layers, in_planes, out_planes, block, stride, dr...
    method _make_layer (line 47) | def _make_layer(self, block, in_planes, out_planes, nb_layers, stride,...
    method forward (line 57) | def forward(self, x):
  class WideResNet (line 61) | class WideResNet(nn.Module):
    method __init__ (line 62) | def __init__(self, depth, num_classes, widen_factor=1, dropout=0.0):
    method forward (line 94) | def forward(self, x):

FILE: pipeline/models/image_models/wide_resnet_fixup.py
  class BasicBlock (line 15) | class BasicBlock(nn.Module):
    method __init__ (line 18) | def __init__(self, in_planes, out_planes, stride, dropout, fixup_l, fi...
    method forward (line 42) | def forward(self, x):
  class NetworkBlock (line 56) | class NetworkBlock(nn.Module):
    method __init__ (line 57) | def __init__(self, nb_layers, in_planes, out_planes, block, stride, dr...
    method _make_layer (line 61) | def _make_layer(self, block, in_planes, out_planes, nb_layers, stride,...
    method forward (line 71) | def forward(self, x):
  class WideResNet (line 75) | class WideResNet(nn.Module):
    method __init__ (line 76) | def __init__(self, depth, num_classes, widen_factor=1, dropout=0.0, fi...
    method forward (line 109) | def forward(self, x):

FILE: pipeline/predictors/base.py
  class PredictorBase (line 14) | class PredictorBase:
    method __init__ (line 15) | def __init__(
    method predict_step (line 31) | def predict_step(self, input_data: torch.Tensor):
    method log_predict_step (line 36) | def log_predict_step(self, step_id: int, predict_time: float):
    method log_predict_completed (line 43) | def log_predict_completed(self, predict_time: float):
    method load_last_model (line 47) | def load_last_model(self):
    method run (line 60) | def run(self):

FILE: pipeline/predictors/classification.py
  class PredictorClassification (line 5) | class PredictorClassification(PredictorBase):
    method predict_step (line 6) | def predict_step(self, input_data: torch.Tensor):

FILE: pipeline/schedulers/base.py
  class SchedulerBase (line 4) | class SchedulerBase(abc.ABC):
    method step (line 6) | def step(self, loss, metrics, epoch_id):
  class SchedulerWrapperBase (line 10) | class SchedulerWrapperBase(SchedulerBase):
    method __init__ (line 11) | def __init__(self, scheduler):
  class SchedulerWrapperIdentity (line 15) | class SchedulerWrapperIdentity(SchedulerWrapperBase):
    method __init__ (line 16) | def __init__(self, *args, **kwargs):
    method step (line 19) | def step(self, loss, metrics, epoch_id):
  class SchedulerWrapperLossBase (line 23) | class SchedulerWrapperLossBase(SchedulerWrapperBase):
    method __init__ (line 24) | def __init__(self, scheduler):
    method step (line 27) | def step(self, loss, metrics, epoch_id):
  class SchedulerWrapperMetricsMeanBase (line 31) | class SchedulerWrapperMetricsMeanBase(SchedulerWrapperBase):
    method __init__ (line 32) | def __init__(self, scheduler):
    method step (line 35) | def step(self, loss, metrics, epoch_id):

FILE: pipeline/schedulers/dropout/increase_step.py
  class SchedulerWrapperIncreaseStep (line 6) | class SchedulerWrapperIncreaseStep(SchedulerBase):
    method __init__ (line 7) | def __init__(self, model, epoch_count, initial_value=0, max_value=0.5):
    method step (line 13) | def step(self, loss, metrics, epoch_id):

FILE: pipeline/schedulers/dropout/utils.py
  function set_dropout_probability (line 5) | def set_dropout_probability(module, probability):

FILE: pipeline/schedulers/learning_rate/cyclical_lr_scheduler.py
  class SchedulerWrapperLossOnCyclic (line 6) | class SchedulerWrapperLossOnCyclic(SchedulerWrapperLossBase):
    method __init__ (line 7) | def __init__(self, optimizer, T_max, eta_min=0, last_epoch=-1):
  class SchedulerWrapperMetricsMeanOnCyclic (line 17) | class SchedulerWrapperMetricsMeanOnCyclic(SchedulerWrapperMetricsMeanBase):
    method __init__ (line 18) | def __init__(self, optimizer, T_max, eta_min=0, last_epoch=-1):

FILE: pipeline/schedulers/learning_rate/reduce_on_plateau.py
  class SchedulerWrapperLossOnPlateau (line 6) | class SchedulerWrapperLossOnPlateau(SchedulerWrapperLossBase):
    method __init__ (line 7) | def __init__(self, optimizer, mode="min", factor=0.5, patience=3, verb...
  class SchedulerWrapperMetricsMeanOnPlateau (line 20) | class SchedulerWrapperMetricsMeanOnPlateau(SchedulerWrapperMetricsMeanBa...
    method __init__ (line 21) | def __init__(self, optimizer, mode="max", factor=0.5, patience=3, verb...

FILE: pipeline/storage/predictions.py
  class PredictionsStorageBase (line 8) | class PredictionsStorageBase(abc.ABC):
    method add (line 10) | def add(self, identifier, prediction):
    method add_batch (line 13) | def add_batch(self, identifiers, predictions):
    method flush (line 18) | def flush(self):
    method get_all (line 22) | def get_all(self):
    method get_by_id (line 26) | def get_by_id(self, identifier):
    method get_by_id_batch (line 29) | def get_by_id_batch(self, identifiers):
    method sort_by_id (line 37) | def sort_by_id(self):
  class PredictionsStorageFiles (line 41) | class PredictionsStorageFiles(PredictionsStorageBase):
    method __init__ (line 42) | def __init__(self, path):
    method _load_predictions (line 58) | def _load_predictions(self):
    method _save_predictions (line 67) | def _save_predictions(self):
    method add (line 76) | def add(self, identifier, prediction):
    method flush (line 81) | def flush(self):
    method get_all (line 84) | def get_all(self):
    method get_by_id (line 87) | def get_by_id(self, identifier):
    method sort_by_id (line 94) | def sort_by_id(self):

FILE: pipeline/storage/state.py
  class StateStorageBase (line 8) | class StateStorageBase(abc.ABC):
    method has_key (line 10) | def has_key(self, key: str):
    method get_value (line 14) | def get_value(self, key: str):
    method remove_key (line 18) | def remove_key(self, key: str):
    method set_value (line 22) | def set_value(self, key: str, value: object):
  class StateStorageEmpty (line 26) | class StateStorageEmpty(StateStorageBase):
    method set_value (line 27) | def set_value(self, key: str, value: object):
    method get_value (line 30) | def get_value(self, key: str):
    method has_key (line 33) | def has_key(self, key: str):
    method remove_key (line 36) | def remove_key(self, key: str):
  class StateStorageFile (line 40) | class StateStorageFile(StateStorageBase):
    method __init__ (line 41) | def __init__(self, path: str):
    method _save (line 52) | def _save(self):
    method has_key (line 56) | def has_key(self, key: str):
    method get_value (line 59) | def get_value(self, key: str):
    method set_value (line 65) | def set_value(self, key: str, value: object):
    method remove_key (line 70) | def remove_key(self, key: str):

FILE: pipeline/trainers/base.py
  class TrainerBase (line 18) | class TrainerBase:
    method __init__ (line 19) | def __init__(
    method train_step (line 47) | def train_step(self, input_data: torch.Tensor, target: torch.Tensor):
    method predict_step (line 62) | def predict_step(self, input_data: torch.Tensor):
    method log_train_step (line 67) | def log_train_step(self, epoch_id: int, step_id: int, epoch_time: floa...
    method log_validation_step (line 75) | def log_validation_step(self, epoch_id: int, step_id: int, epoch_time:...
    method log_train_epoch (line 84) | def log_train_epoch(self, epoch_id: int, epoch_time: float, mean_loss:...
    method log_validation_epoch (line 89) | def log_validation_epoch(self, epoch_id: int, epoch_time: float, mean_...
    method run_train_epoch (line 94) | def run_train_epoch(self, epoch_id: int):
    method run_validation_epoch (line 117) | def run_validation_epoch(self, epoch_id: int):
    method load_optimizer_state (line 146) | def load_optimizer_state(self):
    method save_optimizer_state (line 155) | def save_optimizer_state(self):
    method save_last_model (line 162) | def save_last_model(self, epoch_id):
    method load_last_model (line 168) | def load_last_model(self, epoch_id):
    method run (line 172) | def run(self):

FILE: pipeline/trainers/classification.py
  class TrainerClassification (line 4) | class TrainerClassification(TrainerBase):

FILE: pipeline/trainers/segmentation.py
  class TrainerSegmentation (line 4) | class TrainerSegmentation(TrainerBase):

FILE: pipeline/utils.py
  function _load_cls (line 11) | def _load_cls(module_path, cls_name):
  function load_config (line 23) | def load_config(config_path: str):
  function load_predict_config (line 27) | def load_predict_config(config_path: str):
  function move_to_device (line 31) | def move_to_device(tensor: list or tuple or torch.Tensor, device: str):
  function get_path (line 39) | def get_path(path):
  function save_model (line 43) | def save_model(model, path):
  function load_model (line 51) | def load_model(model, path):
  function run_train (line 58) | def run_train(config):
  function run_predict (line 98) | def run_predict(config):

FILE: tests/common.py
  function make_temp_path (line 5) | def make_temp_path():

FILE: tests/test_metrics.py
  class TestClassificationMetrics (line 8) | class TestClassificationMetrics:
    method test_accuracy (line 9) | def test_accuracy(self):

FILE: tests/test_schedulers.py
  class TestReduceLROnPlateau (line 7) | class TestReduceLROnPlateau:
    method test_wrapper_loss (line 8) | def test_wrapper_loss(self):
    method test_wrapper_metrics (line 64) | def test_wrapper_metrics(self):

FILE: tests/test_storage.py
  class TestStateStorageEmpty (line 9) | class TestStateStorageEmpty:
    method test_set_value (line 10) | def test_set_value(self):
    method test_get_value (line 14) | def test_get_value(self):
    method test_has_key (line 24) | def test_has_key(self):
    method test_remove_key (line 32) | def test_remove_key(self):
  class TestStateStorageFile (line 43) | class TestStateStorageFile:
    method test_basic (line 44) | def test_basic(self):
    method test_save_load (line 56) | def test_save_load(self):