####################
Enot utility package
####################

enot.utils package contains utility functional.


**********
batch_norm
**********

This package contains batch normalization layer utilities. Some examples
are batch norm tuning functionality, resetting all batch norm layers and
checking if module is an instance of PyTorch's BatchNorm class.

.. autofunction:: enot.utils.batch_norm.is_bn

.. autofunction:: enot.utils.batch_norm.reset_bn

.. autofunction:: enot.utils.batch_norm.model_reset_bn

.. autofunction:: enot.utils.batch_norm.model_set_bn_momentum

.. autofunction:: enot.utils.batch_norm.tune_bn_stats


****
data
****

This package contains data-related functional. You can find dataloader
creation functions, dataset classes and image transformation functions.

csv_annotation_dataset
======================

.. autoclass:: enot.utils.data.csv_annotation_dataset.CsvAnnotationDataset
    :members:
    :show-inheritance:

dataloaders
===========

.. autofunction:: enot.utils.data.dataloaders.get_default_train_transform

.. autofunction:: enot.utils.data.dataloaders.get_default_validation_transform

.. autofunction:: enot.utils.data.dataloaders.create_data_loader

.. autofunction:: enot.utils.data.dataloaders.create_data_loader_from_csv_annotation


******
common
******

This package contains regular PyTorch utility functions which can be used
outside architecture optimization pipeline.

.. autofunction:: enot.utils.common.is_floating_tensor

.. autofunction:: enot.utils.common.profile_speed

.. autofunction:: enot.utils.common.save

.. autofunction:: enot.utils.common.load

.. autofunction:: enot.utils.common.init_convnet_params

.. autofunction:: enot.utils.common.iterate_by_submodules

.. autofunction:: enot.utils.common.cast_to_numpy

.. autofunction:: enot.utils.common.replace_ops


*****
train
*****

This package contains train-related utility functions.

.. autofunction:: enot.utils.train.split_into_groups