Thank you! We'll be in touch ASAP.
Something went wrong, please try again or contact us directly at contact@dagshub.com
Deci-AI:master
deci-ai:bugfix/infra-000_ci
<!DOCTYPE html> <html class="writer-html5" lang="en" > <head> <meta charset="utf-8" /> <meta name="viewport" content="width=device-width, initial-scale=1.0" /> <title>super_gradients package — SuperGradients 3.0.3 documentation</title> <link rel="stylesheet" href="_static/pygments.css" type="text/css" /> <link rel="stylesheet" href="_static/css/theme.css" type="text/css" /> <link rel="stylesheet" href="_static/graphviz.css" type="text/css" /> <link rel="stylesheet" href="_static/custom.css" type="text/css" /> <!--[if lt IE 9]> <script src="_static/js/html5shiv.min.js"></script> <![endif]--> <script data-url_root="./" id="documentation_options" src="_static/documentation_options.js"></script> <script src="_static/jquery.js"></script> <script src="_static/underscore.js"></script> <script src="_static/_sphinx_javascript_frameworks_compat.js"></script> <script src="_static/doctools.js"></script> <script src="_static/sphinx_highlight.js"></script> <script src="_static/js/theme.js"></script> <link rel="index" title="Index" href="genindex.html" /> <link rel="search" title="Search" href="search.html" /> </head> <body class="wy-body-for-nav"> <div class="wy-grid-for-nav"> <nav data-toggle="wy-nav-shift" class="wy-nav-side"> <div class="wy-side-scroll"> <div class="wy-side-nav-search" > <a href="index.html" class="icon icon-home"> SuperGradients </a> <div role="search"> <form id="rtd-search-form" class="wy-form" action="search.html" method="get"> <input type="text" name="q" placeholder="Search docs" /> <input type="hidden" name="check_keywords" value="yes" /> <input type="hidden" name="area" value="default" /> </form> </div> </div><div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="Navigation menu"> <p class="caption" role="heading"><span class="caption-text">Welcome To SuperGradients</span></p> <ul> <li class="toctree-l1"><a class="reference internal" href="welcome.html">Version 3 is out! Notebooks have been updated!</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#build-with-supergradients">Build with SuperGradients</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#quick-installation">Quick Installation</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#what-s-new">What’s New</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#coming-soon">Coming soon</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#table-of-content">Table of Content</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#getting-started">Getting Started</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#advanced-features">Advanced Features</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#installation-methods">Installation Methods</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#implemented-model-architectures">Implemented Model Architectures</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#documentation">Documentation</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#contributing">Contributing</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#citation">Citation</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#community">Community</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#license">License</a></li> <li class="toctree-l1"><a class="reference internal" href="welcome.html#deci-platform">Deci Platform</a></li> </ul> <p class="caption" role="heading"><span class="caption-text">Technical Documentation</span></p> <ul> <li class="toctree-l1"><a class="reference internal" href="super_gradients.common.html">Common package</a></li> <li class="toctree-l1"><a class="reference internal" href="super_gradients.training.html">Training package</a></li> </ul> </div> </div> </nav> <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap"><nav class="wy-nav-top" aria-label="Mobile navigation menu" > <i data-toggle="wy-nav-top" class="fa fa-bars"></i> <a href="index.html">SuperGradients</a> </nav> <div class="wy-nav-content"> <div class="rst-content"> <div role="navigation" aria-label="Page navigation"> <ul class="wy-breadcrumbs"> <li><a href="index.html" class="icon icon-home"></a> »</li> <li>super_gradients package</li> <li class="wy-breadcrumbs-aside"> <a href="_sources/super_gradients.rst.txt" rel="nofollow"> View page source</a> </li> </ul> <hr/> </div> <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article"> <div itemprop="articleBody"> <div class="section" id="super-gradients-package"> <h1>super_gradients package<a class="headerlink" href="#super-gradients-package" title="Permalink to this heading"></a></h1> <div class="toctree-wrapper compound"> <ul> <li class="toctree-l1"><a class="reference internal" href="super_gradients.common.html">Common package</a><ul> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.setup_crash_handler"><code class="docutils literal notranslate"><span class="pre">setup_crash_handler()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.explicit_params_validation"><code class="docutils literal notranslate"><span class="pre">explicit_params_validation()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.singleton"><code class="docutils literal notranslate"><span class="pre">singleton()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AWSConnector"><code class="docutils literal notranslate"><span class="pre">AWSConnector</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AWSConnector.get_aws_session"><code class="docutils literal notranslate"><span class="pre">AWSConnector.get_aws_session()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AWSConnector.get_aws_client_for_service_name"><code class="docutils literal notranslate"><span class="pre">AWSConnector.get_aws_client_for_service_name()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AWSConnector.get_aws_resource_for_service_name"><code class="docutils literal notranslate"><span class="pre">AWSConnector.get_aws_resource_for_service_name()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AWSConnector.is_client_error"><code class="docutils literal notranslate"><span class="pre">AWSConnector.is_client_error()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DatasetDataInterface"><code class="docutils literal notranslate"><span class="pre">DatasetDataInterface</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DatasetDataInterface.load_remote_dataset_file"><code class="docutils literal notranslate"><span class="pre">DatasetDataInterface.load_remote_dataset_file()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces.load_all_remote_log_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.load_all_remote_log_files()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces.save_all_remote_checkpoint_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.save_all_remote_checkpoint_files()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces.load_remote_checkpoints_file"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.load_remote_checkpoints_file()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces.load_remote_logging_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.load_remote_logging_files()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces.save_remote_checkpoints_file"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.save_remote_checkpoints_file()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.ADNNModelRepositoryDataInterfaces.save_remote_tensorboard_event_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.save_remote_tensorboard_event_files()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector"><code class="docutils literal notranslate"><span class="pre">S3Connector</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.check_key_exists"><code class="docutils literal notranslate"><span class="pre">S3Connector.check_key_exists()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.get_object_by_etag"><code class="docutils literal notranslate"><span class="pre">S3Connector.get_object_by_etag()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.create_bucket"><code class="docutils literal notranslate"><span class="pre">S3Connector.create_bucket()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.delete_bucket"><code class="docutils literal notranslate"><span class="pre">S3Connector.delete_bucket()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.get_object_metadata"><code class="docutils literal notranslate"><span class="pre">S3Connector.get_object_metadata()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.delete_key"><code class="docutils literal notranslate"><span class="pre">S3Connector.delete_key()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.upload_file_from_stream"><code class="docutils literal notranslate"><span class="pre">S3Connector.upload_file_from_stream()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.upload_file"><code class="docutils literal notranslate"><span class="pre">S3Connector.upload_file()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.download_key"><code class="docutils literal notranslate"><span class="pre">S3Connector.download_key()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.download_keys_by_prefix"><code class="docutils literal notranslate"><span class="pre">S3Connector.download_keys_by_prefix()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.download_file_by_path"><code class="docutils literal notranslate"><span class="pre">S3Connector.download_file_by_path()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.empty_folder_content_by_path_prefix"><code class="docutils literal notranslate"><span class="pre">S3Connector.empty_folder_content_by_path_prefix()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.upload_buffer"><code class="docutils literal notranslate"><span class="pre">S3Connector.upload_buffer()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.list_bucket_objects"><code class="docutils literal notranslate"><span class="pre">S3Connector.list_bucket_objects()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.create_presigned_upload_url"><code class="docutils literal notranslate"><span class="pre">S3Connector.create_presigned_upload_url()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.create_presigned_download_url"><code class="docutils literal notranslate"><span class="pre">S3Connector.create_presigned_download_url()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.convert_content_length_to_mb"><code class="docutils literal notranslate"><span class="pre">S3Connector.convert_content_length_to_mb()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.S3Connector.copy_key"><code class="docutils literal notranslate"><span class="pre">S3Connector.copy_key()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.init_trainer"><code class="docutils literal notranslate"><span class="pre">init_trainer()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.is_distributed"><code class="docutils literal notranslate"><span class="pre">is_distributed()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.StrictLoad"><code class="docutils literal notranslate"><span class="pre">StrictLoad</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.StrictLoad.OFF"><code class="docutils literal notranslate"><span class="pre">StrictLoad.OFF</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.StrictLoad.ON"><code class="docutils literal notranslate"><span class="pre">StrictLoad.ON</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.StrictLoad.NO_KEY_MATCHING"><code class="docutils literal notranslate"><span class="pre">StrictLoad.NO_KEY_MATCHING</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.CLASSIFICATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.CLASSIFICATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.SEMANTIC_SEGMENTATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.SEMANTIC_SEGMENTATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.OBJECT_DETECTION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.OBJECT_DETECTION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.DEPTH_ESTIMATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.DEPTH_ESTIMATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.POSE_ESTIMATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.POSE_ESTIMATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.NLP"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.NLP</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.DeepLearningTask.OTHER"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.OTHER</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.EvaluationType"><code class="docutils literal notranslate"><span class="pre">EvaluationType</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.EvaluationType.TEST"><code class="docutils literal notranslate"><span class="pre">EvaluationType.TEST</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.EvaluationType.VALIDATION"><code class="docutils literal notranslate"><span class="pre">EvaluationType.VALIDATION</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.MultiGPUMode"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.MultiGPUMode.OFF"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.OFF</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.MultiGPUMode.DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DATA_PARALLEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.MultiGPUMode.DISTRIBUTED_DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DISTRIBUTED_DATA_PARALLEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.MultiGPUMode.AUTO"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.AUTO</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.MultiGPUMode.dict"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.dict()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.UpsampleMode"><code class="docutils literal notranslate"><span class="pre">UpsampleMode</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.UpsampleMode.NEAREST"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.NEAREST</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.UpsampleMode.BILINEAR"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.BILINEAR</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.UpsampleMode.BICUBIC"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.BICUBIC</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.UpsampleMode.SNPE_BILINEAR"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.SNPE_BILINEAR</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig.FILE_LOGGING_LEVEL"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.FILE_LOGGING_LEVEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig.CONSOLE_LOGGING_LEVEL"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.CONSOLE_LOGGING_LEVEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig.filename"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.filename</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig.get_instance"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.get_instance()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig.get_log_file_path"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.get_log_file_path()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.AutoLoggerConfig.setup_logging"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.setup_logging()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig.FILE_LOGGING_LEVEL"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.FILE_LOGGING_LEVEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig.CONSOLE_LOGGING_LEVEL"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.CONSOLE_LOGGING_LEVEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig.filename"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.filename</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig.get_instance"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.get_instance()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig.get_log_file_path"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.get_log_file_path()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.AutoLoggerConfig.setup_logging"><code class="docutils literal notranslate"><span class="pre">AutoLoggerConfig.setup_logging()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.ConsoleSink"><code class="docutils literal notranslate"><span class="pre">ConsoleSink</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.ConsoleSink.set_location"><code class="docutils literal notranslate"><span class="pre">ConsoleSink.set_location()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.ConsoleSink.flush"><code class="docutils literal notranslate"><span class="pre">ConsoleSink.flush()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.auto_logging.ConsoleSink.get_filename"><code class="docutils literal notranslate"><span class="pre">ConsoleSink.get_filename()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector"><code class="docutils literal notranslate"><span class="pre">S3Connector</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.check_key_exists"><code class="docutils literal notranslate"><span class="pre">S3Connector.check_key_exists()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.get_object_by_etag"><code class="docutils literal notranslate"><span class="pre">S3Connector.get_object_by_etag()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.create_bucket"><code class="docutils literal notranslate"><span class="pre">S3Connector.create_bucket()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.delete_bucket"><code class="docutils literal notranslate"><span class="pre">S3Connector.delete_bucket()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.get_object_metadata"><code class="docutils literal notranslate"><span class="pre">S3Connector.get_object_metadata()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.delete_key"><code class="docutils literal notranslate"><span class="pre">S3Connector.delete_key()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.upload_file_from_stream"><code class="docutils literal notranslate"><span class="pre">S3Connector.upload_file_from_stream()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.upload_file"><code class="docutils literal notranslate"><span class="pre">S3Connector.upload_file()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.download_key"><code class="docutils literal notranslate"><span class="pre">S3Connector.download_key()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.download_keys_by_prefix"><code class="docutils literal notranslate"><span class="pre">S3Connector.download_keys_by_prefix()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.download_file_by_path"><code class="docutils literal notranslate"><span class="pre">S3Connector.download_file_by_path()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.empty_folder_content_by_path_prefix"><code class="docutils literal notranslate"><span class="pre">S3Connector.empty_folder_content_by_path_prefix()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.upload_buffer"><code class="docutils literal notranslate"><span class="pre">S3Connector.upload_buffer()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.list_bucket_objects"><code class="docutils literal notranslate"><span class="pre">S3Connector.list_bucket_objects()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.create_presigned_upload_url"><code class="docutils literal notranslate"><span class="pre">S3Connector.create_presigned_upload_url()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.create_presigned_download_url"><code class="docutils literal notranslate"><span class="pre">S3Connector.create_presigned_download_url()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.convert_content_length_to_mb"><code class="docutils literal notranslate"><span class="pre">S3Connector.convert_content_length_to_mb()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_connection.S3Connector.copy_key"><code class="docutils literal notranslate"><span class="pre">S3Connector.copy_key()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.DatasetDataInterface"><code class="docutils literal notranslate"><span class="pre">DatasetDataInterface</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.DatasetDataInterface.load_remote_dataset_file"><code class="docutils literal notranslate"><span class="pre">DatasetDataInterface.load_remote_dataset_file()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces.load_all_remote_log_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.load_all_remote_log_files()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces.save_all_remote_checkpoint_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.save_all_remote_checkpoint_files()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces.load_remote_checkpoints_file"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.load_remote_checkpoints_file()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces.load_remote_logging_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.load_remote_logging_files()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces.save_remote_checkpoints_file"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.save_remote_checkpoints_file()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_interface.ADNNModelRepositoryDataInterfaces.save_remote_tensorboard_event_files"><code class="docutils literal notranslate"><span class="pre">ADNNModelRepositoryDataInterfaces.save_remote_tensorboard_event_files()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.StrictLoad"><code class="docutils literal notranslate"><span class="pre">StrictLoad</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.StrictLoad.OFF"><code class="docutils literal notranslate"><span class="pre">StrictLoad.OFF</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.StrictLoad.ON"><code class="docutils literal notranslate"><span class="pre">StrictLoad.ON</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.StrictLoad.NO_KEY_MATCHING"><code class="docutils literal notranslate"><span class="pre">StrictLoad.NO_KEY_MATCHING</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.CLASSIFICATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.CLASSIFICATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.SEMANTIC_SEGMENTATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.SEMANTIC_SEGMENTATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.OBJECT_DETECTION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.OBJECT_DETECTION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.DEPTH_ESTIMATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.DEPTH_ESTIMATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.POSE_ESTIMATION"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.POSE_ESTIMATION</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.NLP"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.NLP</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.DeepLearningTask.OTHER"><code class="docutils literal notranslate"><span class="pre">DeepLearningTask.OTHER</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.EvaluationType"><code class="docutils literal notranslate"><span class="pre">EvaluationType</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.EvaluationType.TEST"><code class="docutils literal notranslate"><span class="pre">EvaluationType.TEST</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.EvaluationType.VALIDATION"><code class="docutils literal notranslate"><span class="pre">EvaluationType.VALIDATION</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.MultiGPUMode"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.MultiGPUMode.OFF"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.OFF</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.MultiGPUMode.DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DATA_PARALLEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.MultiGPUMode.DISTRIBUTED_DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DISTRIBUTED_DATA_PARALLEL</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.MultiGPUMode.AUTO"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.AUTO</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.MultiGPUMode.dict"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.dict()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.UpsampleMode"><code class="docutils literal notranslate"><span class="pre">UpsampleMode</span></code></a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.UpsampleMode.NEAREST"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.NEAREST</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.UpsampleMode.BILINEAR"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.BILINEAR</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.UpsampleMode.BICUBIC"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.BICUBIC</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.data_types.UpsampleMode.SNPE_BILINEAR"><code class="docutils literal notranslate"><span class="pre">UpsampleMode.SNPE_BILINEAR</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.decorators.explicit_params_validation"><code class="docutils literal notranslate"><span class="pre">explicit_params_validation()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.decorators.singleton"><code class="docutils literal notranslate"><span class="pre">singleton()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.environment.init_trainer"><code class="docutils literal notranslate"><span class="pre">init_trainer()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.environment.is_distributed"><code class="docutils literal notranslate"><span class="pre">is_distributed()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.registry.register_model"><code class="docutils literal notranslate"><span class="pre">register_model()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.registry.register_detection_module"><code class="docutils literal notranslate"><span class="pre">register_detection_module()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.registry.register_metric"><code class="docutils literal notranslate"><span class="pre">register_metric()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#super_gradients.common.registry.register_loss"><code class="docutils literal notranslate"><span class="pre">register_loss()</span></code></a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.common.html#module-contents">Module contents</a></li> </ul> </li> <li class="toctree-l1"><a class="reference internal" href="super_gradients.training.html">Training package</a><ul> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training">super_gradients.training module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.DataAugmentation"><code class="docutils literal notranslate"><span class="pre">DataAugmentation</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.DataAugmentation.to_tensor"><code class="docutils literal notranslate"><span class="pre">DataAugmentation.to_tensor()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.DataAugmentation.normalize"><code class="docutils literal notranslate"><span class="pre">DataAugmentation.normalize()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.DataAugmentation.cutout"><code class="docutils literal notranslate"><span class="pre">DataAugmentation.cutout()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer"><code class="docutils literal notranslate"><span class="pre">Trainer</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.train"><code class="docutils literal notranslate"><span class="pre">Trainer.train()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.predict"><code class="docutils literal notranslate"><span class="pre">Trainer.predict()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.train_from_config"><code class="docutils literal notranslate"><span class="pre">Trainer.train_from_config()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.resume_experiment"><code class="docutils literal notranslate"><span class="pre">Trainer.resume_experiment()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.evaluate_from_recipe"><code class="docutils literal notranslate"><span class="pre">Trainer.evaluate_from_recipe()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.evaluate_checkpoint"><code class="docutils literal notranslate"><span class="pre">Trainer.evaluate_checkpoint()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#id0"><code class="docutils literal notranslate"><span class="pre">Trainer.train()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.get_arch_params"><code class="docutils literal notranslate"><span class="pre">Trainer.get_arch_params</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.get_structure"><code class="docutils literal notranslate"><span class="pre">Trainer.get_structure</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.get_architecture"><code class="docutils literal notranslate"><span class="pre">Trainer.get_architecture</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.set_experiment_name"><code class="docutils literal notranslate"><span class="pre">Trainer.set_experiment_name()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.get_module"><code class="docutils literal notranslate"><span class="pre">Trainer.get_module</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.set_module"><code class="docutils literal notranslate"><span class="pre">Trainer.set_module()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.test"><code class="docutils literal notranslate"><span class="pre">Trainer.test()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.evaluate"><code class="docutils literal notranslate"><span class="pre">Trainer.evaluate()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.get_net"><code class="docutils literal notranslate"><span class="pre">Trainer.get_net</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.set_net"><code class="docutils literal notranslate"><span class="pre">Trainer.set_net()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.set_ckpt_best_name"><code class="docutils literal notranslate"><span class="pre">Trainer.set_ckpt_best_name()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.Trainer.set_ema"><code class="docutils literal notranslate"><span class="pre">Trainer.set_ema()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.KDTrainer"><code class="docutils literal notranslate"><span class="pre">KDTrainer</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.KDTrainer.train_from_config"><code class="docutils literal notranslate"><span class="pre">KDTrainer.train_from_config()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.KDTrainer.train"><code class="docutils literal notranslate"><span class="pre">KDTrainer.train()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.MultiGPUMode"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.MultiGPUMode.OFF"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.OFF</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.MultiGPUMode.DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DATA_PARALLEL</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.MultiGPUMode.DISTRIBUTED_DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DISTRIBUTED_DATA_PARALLEL</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.MultiGPUMode.AUTO"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.AUTO</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.MultiGPUMode.dict"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.dict()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.StrictLoad"><code class="docutils literal notranslate"><span class="pre">StrictLoad</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.StrictLoad.OFF"><code class="docutils literal notranslate"><span class="pre">StrictLoad.OFF</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.StrictLoad.ON"><code class="docutils literal notranslate"><span class="pre">StrictLoad.ON</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.StrictLoad.NO_KEY_MATCHING"><code class="docutils literal notranslate"><span class="pre">StrictLoad.NO_KEY_MATCHING</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.EvaluationType"><code class="docutils literal notranslate"><span class="pre">EvaluationType</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.EvaluationType.TEST"><code class="docutils literal notranslate"><span class="pre">EvaluationType.TEST</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.EvaluationType.VALIDATION"><code class="docutils literal notranslate"><span class="pre">EvaluationType.VALIDATION</span></code></a></li> </ul> </li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#super-gradients-training-datasets-module">super_gradients.training.datasets module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DataAugmentation"><code class="docutils literal notranslate"><span class="pre">DataAugmentation</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DataAugmentation.to_tensor"><code class="docutils literal notranslate"><span class="pre">DataAugmentation.to_tensor()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DataAugmentation.normalize"><code class="docutils literal notranslate"><span class="pre">DataAugmentation.normalize()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DataAugmentation.cutout"><code class="docutils literal notranslate"><span class="pre">DataAugmentation.cutout()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.ListDataset"><code class="docutils literal notranslate"><span class="pre">ListDataset</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DirectoryDataSet"><code class="docutils literal notranslate"><span class="pre">DirectoryDataSet</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SegmentationDataSet"><code class="docutils literal notranslate"><span class="pre">SegmentationDataSet</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SegmentationDataSet.sample_loader"><code class="docutils literal notranslate"><span class="pre">SegmentationDataSet.sample_loader()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SegmentationDataSet.sample_transform"><code class="docutils literal notranslate"><span class="pre">SegmentationDataSet.sample_transform()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SegmentationDataSet.target_loader"><code class="docutils literal notranslate"><span class="pre">SegmentationDataSet.target_loader()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SegmentationDataSet.target_transform"><code class="docutils literal notranslate"><span class="pre">SegmentationDataSet.target_transform()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOC2012SegmentationDataSet"><code class="docutils literal notranslate"><span class="pre">PascalVOC2012SegmentationDataSet</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOC2012SegmentationDataSet.IGNORE_LABEL"><code class="docutils literal notranslate"><span class="pre">PascalVOC2012SegmentationDataSet.IGNORE_LABEL</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOC2012SegmentationDataSet.target_transform"><code class="docutils literal notranslate"><span class="pre">PascalVOC2012SegmentationDataSet.target_transform()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOC2012SegmentationDataSet.decode_segmentation_mask"><code class="docutils literal notranslate"><span class="pre">PascalVOC2012SegmentationDataSet.decode_segmentation_mask()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalAUG2012SegmentationDataSet"><code class="docutils literal notranslate"><span class="pre">PascalAUG2012SegmentationDataSet</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalAUG2012SegmentationDataSet.target_loader"><code class="docutils literal notranslate"><span class="pre">PascalAUG2012SegmentationDataSet.target_loader()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOCAndAUGUnifiedDataset"><code class="docutils literal notranslate"><span class="pre">PascalVOCAndAUGUnifiedDataset</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOCAndAUGUnifiedDataset.datasets"><code class="docutils literal notranslate"><span class="pre">PascalVOCAndAUGUnifiedDataset.datasets</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOCAndAUGUnifiedDataset.cumulative_sizes"><code class="docutils literal notranslate"><span class="pre">PascalVOCAndAUGUnifiedDataset.cumulative_sizes</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.CoCoSegmentationDataSet"><code class="docutils literal notranslate"><span class="pre">CoCoSegmentationDataSet</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.CoCoSegmentationDataSet.target_loader"><code class="docutils literal notranslate"><span class="pre">CoCoSegmentationDataSet.target_loader()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset"><code class="docutils literal notranslate"><span class="pre">DetectionDataset</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.get_random_item"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.get_random_item()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.get_sample"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.get_sample()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.get_resized_image"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.get_resized_image()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.apply_transforms"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.apply_transforms()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.get_random_samples"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.get_random_samples()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.get_random_sample"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.get_random_sample()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.output_target_format"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.output_target_format</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.DetectionDataset.plot"><code class="docutils literal notranslate"><span class="pre">DetectionDataset.plot()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.COCODetectionDataset"><code class="docutils literal notranslate"><span class="pre">COCODetectionDataset</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOCDetectionDataset"><code class="docutils literal notranslate"><span class="pre">PascalVOCDetectionDataset</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.PascalVOCDetectionDataset.download"><code class="docutils literal notranslate"><span class="pre">PascalVOCDetectionDataset.download()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.ImageNetDataset"><code class="docutils literal notranslate"><span class="pre">ImageNetDataset</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.Cifar10"><code class="docutils literal notranslate"><span class="pre">Cifar10</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.Cifar100"><code class="docutils literal notranslate"><span class="pre">Cifar100</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SuperviselyPersonsDataset"><code class="docutils literal notranslate"><span class="pre">SuperviselyPersonsDataset</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.datasets.SuperviselyPersonsDataset.CLASS_LABELS"><code class="docutils literal notranslate"><span class="pre">SuperviselyPersonsDataset.CLASS_LABELS</span></code></a></li> </ul> </li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#super-gradients-training-dataloaders-module">super_gradients.training.dataloaders module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco2017_train"><code class="docutils literal notranslate"><span class="pre">coco2017_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco2017_val"><code class="docutils literal notranslate"><span class="pre">coco2017_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco2017_train_yolox"><code class="docutils literal notranslate"><span class="pre">coco2017_train_yolox()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco2017_val_yolox"><code class="docutils literal notranslate"><span class="pre">coco2017_val_yolox()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco2017_train_ssd_lite_mobilenet_v2"><code class="docutils literal notranslate"><span class="pre">coco2017_train_ssd_lite_mobilenet_v2()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco2017_val_ssd_lite_mobilenet_v2"><code class="docutils literal notranslate"><span class="pre">coco2017_val_ssd_lite_mobilenet_v2()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_train"><code class="docutils literal notranslate"><span class="pre">imagenet_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_val"><code class="docutils literal notranslate"><span class="pre">imagenet_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_efficientnet_train"><code class="docutils literal notranslate"><span class="pre">imagenet_efficientnet_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_efficientnet_val"><code class="docutils literal notranslate"><span class="pre">imagenet_efficientnet_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_mobilenetv2_train"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv2_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_mobilenetv2_val"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv2_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_mobilenetv3_train"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv3_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_mobilenetv3_val"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv3_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_regnetY_train"><code class="docutils literal notranslate"><span class="pre">imagenet_regnetY_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_regnetY_val"><code class="docutils literal notranslate"><span class="pre">imagenet_regnetY_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_resnet50_train"><code class="docutils literal notranslate"><span class="pre">imagenet_resnet50_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_resnet50_val"><code class="docutils literal notranslate"><span class="pre">imagenet_resnet50_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_resnet50_kd_train"><code class="docutils literal notranslate"><span class="pre">imagenet_resnet50_kd_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_resnet50_kd_val"><code class="docutils literal notranslate"><span class="pre">imagenet_resnet50_kd_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_vit_base_train"><code class="docutils literal notranslate"><span class="pre">imagenet_vit_base_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.imagenet_vit_base_val"><code class="docutils literal notranslate"><span class="pre">imagenet_vit_base_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.tiny_imagenet_train"><code class="docutils literal notranslate"><span class="pre">tiny_imagenet_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.tiny_imagenet_val"><code class="docutils literal notranslate"><span class="pre">tiny_imagenet_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cifar10_train"><code class="docutils literal notranslate"><span class="pre">cifar10_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cifar10_val"><code class="docutils literal notranslate"><span class="pre">cifar10_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cifar100_train"><code class="docutils literal notranslate"><span class="pre">cifar100_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cifar100_val"><code class="docutils literal notranslate"><span class="pre">cifar100_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_train"><code class="docutils literal notranslate"><span class="pre">cityscapes_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_val"><code class="docutils literal notranslate"><span class="pre">cityscapes_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_stdc_seg50_train"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_seg50_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_stdc_seg50_val"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_seg50_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_stdc_seg75_train"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_seg75_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_stdc_seg75_val"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_seg75_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_regseg48_train"><code class="docutils literal notranslate"><span class="pre">cityscapes_regseg48_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_regseg48_val"><code class="docutils literal notranslate"><span class="pre">cityscapes_regseg48_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_ddrnet_train"><code class="docutils literal notranslate"><span class="pre">cityscapes_ddrnet_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.cityscapes_ddrnet_val"><code class="docutils literal notranslate"><span class="pre">cityscapes_ddrnet_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco_segmentation_train"><code class="docutils literal notranslate"><span class="pre">coco_segmentation_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.coco_segmentation_val"><code class="docutils literal notranslate"><span class="pre">coco_segmentation_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.pascal_aug_segmentation_train"><code class="docutils literal notranslate"><span class="pre">pascal_aug_segmentation_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.pascal_aug_segmentation_val"><code class="docutils literal notranslate"><span class="pre">pascal_aug_segmentation_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.pascal_voc_segmentation_train"><code class="docutils literal notranslate"><span class="pre">pascal_voc_segmentation_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.pascal_voc_segmentation_val"><code class="docutils literal notranslate"><span class="pre">pascal_voc_segmentation_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.supervisely_persons_train"><code class="docutils literal notranslate"><span class="pre">supervisely_persons_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.supervisely_persons_val"><code class="docutils literal notranslate"><span class="pre">supervisely_persons_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.pascal_voc_detection_train"><code class="docutils literal notranslate"><span class="pre">pascal_voc_detection_train()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.pascal_voc_detection_val"><code class="docutils literal notranslate"><span class="pre">pascal_voc_detection_val()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.get_data_loader"><code class="docutils literal notranslate"><span class="pre">get_data_loader()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.dataloaders.get"><code class="docutils literal notranslate"><span class="pre">get()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#super-gradients-training-exceptions-module">super_gradients.training.exceptions module</a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#super-gradients-training-kd-trainer-module">super_gradients.training.kd_trainer module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.kd_trainer.KDTrainer"><code class="docutils literal notranslate"><span class="pre">KDTrainer</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.kd_trainer.KDTrainer.train_from_config"><code class="docutils literal notranslate"><span class="pre">KDTrainer.train_from_config()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.kd_trainer.KDTrainer.train"><code class="docutils literal notranslate"><span class="pre">KDTrainer.train()</span></code></a></li> </ul> </li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training.legacy">super_gradients.training.legacy module</a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training.losses">super_gradients.training.losses_models module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses"><code class="docutils literal notranslate"><span class="pre">Losses</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.CROSS_ENTROPY"><code class="docutils literal notranslate"><span class="pre">Losses.CROSS_ENTROPY</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.MSE"><code class="docutils literal notranslate"><span class="pre">Losses.MSE</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.R_SQUARED_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.R_SQUARED_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.SHELFNET_OHEM_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.SHELFNET_OHEM_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.SHELFNET_SE_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.SHELFNET_SE_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.YOLOX_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.YOLOX_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.YOLOX_FAST_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.YOLOX_FAST_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.SSD_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.SSD_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.STDC_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.STDC_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.BCE_DICE_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.BCE_DICE_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.KD_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.KD_LOSS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.Losses.DICE_CE_EDGE_LOSS"><code class="docutils literal notranslate"><span class="pre">Losses.DICE_CE_EDGE_LOSS</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.FocalLoss"><code class="docutils literal notranslate"><span class="pre">FocalLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.FocalLoss.reduction"><code class="docutils literal notranslate"><span class="pre">FocalLoss.reduction</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.FocalLoss.forward"><code class="docutils literal notranslate"><span class="pre">FocalLoss.forward()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.LabelSmoothingCrossEntropyLoss"><code class="docutils literal notranslate"><span class="pre">LabelSmoothingCrossEntropyLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.LabelSmoothingCrossEntropyLoss.forward"><code class="docutils literal notranslate"><span class="pre">LabelSmoothingCrossEntropyLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.LabelSmoothingCrossEntropyLoss.ignore_index"><code class="docutils literal notranslate"><span class="pre">LabelSmoothingCrossEntropyLoss.ignore_index</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.LabelSmoothingCrossEntropyLoss.label_smoothing"><code class="docutils literal notranslate"><span class="pre">LabelSmoothingCrossEntropyLoss.label_smoothing</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetOHEMLoss"><code class="docutils literal notranslate"><span class="pre">ShelfNetOHEMLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetOHEMLoss.forward"><code class="docutils literal notranslate"><span class="pre">ShelfNetOHEMLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetOHEMLoss.component_names"><code class="docutils literal notranslate"><span class="pre">ShelfNetOHEMLoss.component_names</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetOHEMLoss.reduction"><code class="docutils literal notranslate"><span class="pre">ShelfNetOHEMLoss.reduction</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetSemanticEncodingLoss"><code class="docutils literal notranslate"><span class="pre">ShelfNetSemanticEncodingLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetSemanticEncodingLoss.forward"><code class="docutils literal notranslate"><span class="pre">ShelfNetSemanticEncodingLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetSemanticEncodingLoss.component_names"><code class="docutils literal notranslate"><span class="pre">ShelfNetSemanticEncodingLoss.component_names</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetSemanticEncodingLoss.ignore_index"><code class="docutils literal notranslate"><span class="pre">ShelfNetSemanticEncodingLoss.ignore_index</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.ShelfNetSemanticEncodingLoss.label_smoothing"><code class="docutils literal notranslate"><span class="pre">ShelfNetSemanticEncodingLoss.label_smoothing</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.strides"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.strides</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.num_classes"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.num_classes</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.use_l1"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.use_l1</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.center_sampling_radius"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.center_sampling_radius</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.iou_type"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.iou_type</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.component_names"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.component_names</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.forward"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.prepare_predictions"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.prepare_predictions()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.get_l1_target"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.get_l1_target()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.get_assignments"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.get_assignments()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.get_in_boxes_info"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.get_in_boxes_info()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.dynamic_k_matching"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.dynamic_k_matching()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXDetectionLoss.reduction"><code class="docutils literal notranslate"><span class="pre">YoloXDetectionLoss.reduction</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXFastDetectionLoss"><code class="docutils literal notranslate"><span class="pre">YoloXFastDetectionLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXFastDetectionLoss.reduction"><code class="docutils literal notranslate"><span class="pre">YoloXFastDetectionLoss.reduction</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.YoloXFastDetectionLoss.training"><code class="docutils literal notranslate"><span class="pre">YoloXFastDetectionLoss.training</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.RSquaredLoss"><code class="docutils literal notranslate"><span class="pre">RSquaredLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.RSquaredLoss.forward"><code class="docutils literal notranslate"><span class="pre">RSquaredLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.RSquaredLoss.reduction"><code class="docutils literal notranslate"><span class="pre">RSquaredLoss.reduction</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.SSDLoss"><code class="docutils literal notranslate"><span class="pre">SSDLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.SSDLoss.component_names"><code class="docutils literal notranslate"><span class="pre">SSDLoss.component_names</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.SSDLoss.match_dboxes"><code class="docutils literal notranslate"><span class="pre">SSDLoss.match_dboxes()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.SSDLoss.forward"><code class="docutils literal notranslate"><span class="pre">SSDLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.SSDLoss.reduction"><code class="docutils literal notranslate"><span class="pre">SSDLoss.reduction</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.BCEDiceLoss"><code class="docutils literal notranslate"><span class="pre">BCEDiceLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.BCEDiceLoss.loss_weights"><code class="docutils literal notranslate"><span class="pre">BCEDiceLoss.loss_weights</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.BCEDiceLoss.forward"><code class="docutils literal notranslate"><span class="pre">BCEDiceLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.BCEDiceLoss.training"><code class="docutils literal notranslate"><span class="pre">BCEDiceLoss.training</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.KDLogitsLoss"><code class="docutils literal notranslate"><span class="pre">KDLogitsLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.KDLogitsLoss.component_names"><code class="docutils literal notranslate"><span class="pre">KDLogitsLoss.component_names</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.KDLogitsLoss.forward"><code class="docutils literal notranslate"><span class="pre">KDLogitsLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.KDLogitsLoss.reduction"><code class="docutils literal notranslate"><span class="pre">KDLogitsLoss.reduction</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.DiceCEEdgeLoss"><code class="docutils literal notranslate"><span class="pre">DiceCEEdgeLoss</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.DiceCEEdgeLoss.component_names"><code class="docutils literal notranslate"><span class="pre">DiceCEEdgeLoss.component_names</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.DiceCEEdgeLoss.forward"><code class="docutils literal notranslate"><span class="pre">DiceCEEdgeLoss.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.losses.DiceCEEdgeLoss.reduction"><code class="docutils literal notranslate"><span class="pre">DiceCEEdgeLoss.reduction</span></code></a></li> </ul> </li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training.metrics">super_gradients.training.metrics module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics"><code class="docutils literal notranslate"><span class="pre">Metrics</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.ACCURACY"><code class="docutils literal notranslate"><span class="pre">Metrics.ACCURACY</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.TOP5"><code class="docutils literal notranslate"><span class="pre">Metrics.TOP5</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.DETECTION_METRICS"><code class="docutils literal notranslate"><span class="pre">Metrics.DETECTION_METRICS</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.DETECTION_METRICS_050_095"><code class="docutils literal notranslate"><span class="pre">Metrics.DETECTION_METRICS_050_095</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.DETECTION_METRICS_050"><code class="docutils literal notranslate"><span class="pre">Metrics.DETECTION_METRICS_050</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.DETECTION_METRICS_075"><code class="docutils literal notranslate"><span class="pre">Metrics.DETECTION_METRICS_075</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.IOU"><code class="docutils literal notranslate"><span class="pre">Metrics.IOU</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.BINARY_IOU"><code class="docutils literal notranslate"><span class="pre">Metrics.BINARY_IOU</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.DICE"><code class="docutils literal notranslate"><span class="pre">Metrics.DICE</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.BINARY_DICE"><code class="docutils literal notranslate"><span class="pre">Metrics.BINARY_DICE</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Metrics.PIXEL_ACCURACY"><code class="docutils literal notranslate"><span class="pre">Metrics.PIXEL_ACCURACY</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.accuracy"><code class="docutils literal notranslate"><span class="pre">accuracy()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Accuracy"><code class="docutils literal notranslate"><span class="pre">Accuracy</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Accuracy.update"><code class="docutils literal notranslate"><span class="pre">Accuracy.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Accuracy.correct"><code class="docutils literal notranslate"><span class="pre">Accuracy.correct</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Accuracy.total"><code class="docutils literal notranslate"><span class="pre">Accuracy.total</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Top5"><code class="docutils literal notranslate"><span class="pre">Top5</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Top5.update"><code class="docutils literal notranslate"><span class="pre">Top5.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Top5.compute"><code class="docutils literal notranslate"><span class="pre">Top5.compute()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.ToyTestClassificationMetric"><code class="docutils literal notranslate"><span class="pre">ToyTestClassificationMetric</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.ToyTestClassificationMetric.update"><code class="docutils literal notranslate"><span class="pre">ToyTestClassificationMetric.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.ToyTestClassificationMetric.compute"><code class="docutils literal notranslate"><span class="pre">ToyTestClassificationMetric.compute()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.num_cls"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.num_cls</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.post_prediction_callback"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.post_prediction_callback</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.normalize_targets"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.normalize_targets</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.iou_thresholds"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.iou_thresholds</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.recall_thresholds"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.recall_thresholds</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.score_threshold"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.score_threshold</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.top_k_predictions"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.top_k_predictions</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.dist_sync_on_step"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.dist_sync_on_step</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.update"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics.compute"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics.compute()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.PreprocessSegmentationMetricsArgs"><code class="docutils literal notranslate"><span class="pre">PreprocessSegmentationMetricsArgs</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.PixelAccuracy"><code class="docutils literal notranslate"><span class="pre">PixelAccuracy</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.PixelAccuracy.update"><code class="docutils literal notranslate"><span class="pre">PixelAccuracy.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.PixelAccuracy.compute"><code class="docutils literal notranslate"><span class="pre">PixelAccuracy.compute()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.IoU"><code class="docutils literal notranslate"><span class="pre">IoU</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.IoU.update"><code class="docutils literal notranslate"><span class="pre">IoU.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.IoU.confmat"><code class="docutils literal notranslate"><span class="pre">IoU.confmat</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Dice"><code class="docutils literal notranslate"><span class="pre">Dice</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Dice.update"><code class="docutils literal notranslate"><span class="pre">Dice.update()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Dice.compute"><code class="docutils literal notranslate"><span class="pre">Dice.compute()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.Dice.confmat"><code class="docutils literal notranslate"><span class="pre">Dice.confmat</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryIOU"><code class="docutils literal notranslate"><span class="pre">BinaryIOU</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryIOU.compute"><code class="docutils literal notranslate"><span class="pre">BinaryIOU.compute()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryIOU.confmat"><code class="docutils literal notranslate"><span class="pre">BinaryIOU.confmat</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryIOU.training"><code class="docutils literal notranslate"><span class="pre">BinaryIOU.training</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryDice"><code class="docutils literal notranslate"><span class="pre">BinaryDice</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryDice.compute"><code class="docutils literal notranslate"><span class="pre">BinaryDice.compute()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryDice.confmat"><code class="docutils literal notranslate"><span class="pre">BinaryDice.confmat</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.BinaryDice.training"><code class="docutils literal notranslate"><span class="pre">BinaryDice.training</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics_050"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics_050</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics_075"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics_075</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.metrics.DetectionMetrics_050_095"><code class="docutils literal notranslate"><span class="pre">DetectionMetrics_050_095</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training.models">super_gradients.training.models module</a></li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training.sg_trainer">super_gradients.training.sg_model module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer"><code class="docutils literal notranslate"><span class="pre">Trainer</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.train"><code class="docutils literal notranslate"><span class="pre">Trainer.train()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.predict"><code class="docutils literal notranslate"><span class="pre">Trainer.predict()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.train_from_config"><code class="docutils literal notranslate"><span class="pre">Trainer.train_from_config()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.resume_experiment"><code class="docutils literal notranslate"><span class="pre">Trainer.resume_experiment()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.evaluate_from_recipe"><code class="docutils literal notranslate"><span class="pre">Trainer.evaluate_from_recipe()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.evaluate_checkpoint"><code class="docutils literal notranslate"><span class="pre">Trainer.evaluate_checkpoint()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#id3"><code class="docutils literal notranslate"><span class="pre">Trainer.train()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.get_arch_params"><code class="docutils literal notranslate"><span class="pre">Trainer.get_arch_params</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.get_structure"><code class="docutils literal notranslate"><span class="pre">Trainer.get_structure</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.get_architecture"><code class="docutils literal notranslate"><span class="pre">Trainer.get_architecture</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.set_experiment_name"><code class="docutils literal notranslate"><span class="pre">Trainer.set_experiment_name()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.get_module"><code class="docutils literal notranslate"><span class="pre">Trainer.get_module</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.set_module"><code class="docutils literal notranslate"><span class="pre">Trainer.set_module()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.test"><code class="docutils literal notranslate"><span class="pre">Trainer.test()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.evaluate"><code class="docutils literal notranslate"><span class="pre">Trainer.evaluate()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.get_net"><code class="docutils literal notranslate"><span class="pre">Trainer.get_net</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.set_net"><code class="docutils literal notranslate"><span class="pre">Trainer.set_net()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.set_ckpt_best_name"><code class="docutils literal notranslate"><span class="pre">Trainer.set_ckpt_best_name()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.Trainer.set_ema"><code class="docutils literal notranslate"><span class="pre">Trainer.set_ema()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.MultiGPUMode"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.MultiGPUMode.OFF"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.OFF</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.MultiGPUMode.DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DATA_PARALLEL</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.MultiGPUMode.DISTRIBUTED_DATA_PARALLEL"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.DISTRIBUTED_DATA_PARALLEL</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.MultiGPUMode.AUTO"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.AUTO</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.MultiGPUMode.dict"><code class="docutils literal notranslate"><span class="pre">MultiGPUMode.dict()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.StrictLoad"><code class="docutils literal notranslate"><span class="pre">StrictLoad</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.StrictLoad.OFF"><code class="docutils literal notranslate"><span class="pre">StrictLoad.OFF</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.StrictLoad.ON"><code class="docutils literal notranslate"><span class="pre">StrictLoad.ON</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.sg_trainer.StrictLoad.NO_KEY_MATCHING"><code class="docutils literal notranslate"><span class="pre">StrictLoad.NO_KEY_MATCHING</span></code></a></li> </ul> </li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#super-gradients-training-training-hyperparams-module">super_gradients.training.training_hyperparams module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.cifar10_resnet_train_params"><code class="docutils literal notranslate"><span class="pre">cifar10_resnet_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.cityscapes_ddrnet_train_params"><code class="docutils literal notranslate"><span class="pre">cityscapes_ddrnet_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.cityscapes_regseg48_train_params"><code class="docutils literal notranslate"><span class="pre">cityscapes_regseg48_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.cityscapes_stdc_base_train_params"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_base_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.cityscapes_stdc_seg50_train_params"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_seg50_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.cityscapes_stdc_seg75_train_params"><code class="docutils literal notranslate"><span class="pre">cityscapes_stdc_seg75_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.coco2017_ssd_lite_mobilenet_v2_train_params"><code class="docutils literal notranslate"><span class="pre">coco2017_ssd_lite_mobilenet_v2_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.coco2017_yolox_train_params"><code class="docutils literal notranslate"><span class="pre">coco2017_yolox_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.coco_segmentation_shelfnet_lw_train_params"><code class="docutils literal notranslate"><span class="pre">coco_segmentation_shelfnet_lw_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_efficientnet_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_efficientnet_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_mobilenetv2_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv2_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_mobilenetv3_base_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv3_base_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_mobilenetv3_large_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv3_large_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_mobilenetv3_small_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_mobilenetv3_small_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_regnetY_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_regnetY_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_repvgg_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_repvgg_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_resnet50_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_resnet50_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_resnet50_kd_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_resnet50_kd_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_vit_base_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_vit_base_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.imagenet_vit_large_train_params"><code class="docutils literal notranslate"><span class="pre">imagenet_vit_large_train_params()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.training_hyperparams.get"><code class="docutils literal notranslate"><span class="pre">get()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#super-gradients-training-transforms-module">super_gradients.training.transforms module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms"><code class="docutils literal notranslate"><span class="pre">Transforms</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegRandomFlip"><code class="docutils literal notranslate"><span class="pre">Transforms.SegRandomFlip</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegResize"><code class="docutils literal notranslate"><span class="pre">Transforms.SegResize</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegRescale"><code class="docutils literal notranslate"><span class="pre">Transforms.SegRescale</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegRandomRescale"><code class="docutils literal notranslate"><span class="pre">Transforms.SegRandomRescale</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegRandomRotate"><code class="docutils literal notranslate"><span class="pre">Transforms.SegRandomRotate</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegCropImageAndMask"><code class="docutils literal notranslate"><span class="pre">Transforms.SegCropImageAndMask</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegRandomGaussianBlur"><code class="docutils literal notranslate"><span class="pre">Transforms.SegRandomGaussianBlur</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegPadShortToCropSize"><code class="docutils literal notranslate"><span class="pre">Transforms.SegPadShortToCropSize</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.SegColorJitter"><code class="docutils literal notranslate"><span class="pre">Transforms.SegColorJitter</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionMosaic"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionMosaic</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionRandomAffine"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionRandomAffine</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionMixup"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionMixup</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionHSV"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionHSV</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionHorizontalFlip"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionHorizontalFlip</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionPaddedRescale"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionPaddedRescale</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionTargetsFormat"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionTargetsFormat</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.DetectionTargetsFormatTransform"><code class="docutils literal notranslate"><span class="pre">Transforms.DetectionTargetsFormatTransform</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomResizedCropAndInterpolation"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomResizedCropAndInterpolation</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandAugmentTransform"><code class="docutils literal notranslate"><span class="pre">Transforms.RandAugmentTransform</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Lighting"><code class="docutils literal notranslate"><span class="pre">Transforms.Lighting</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomErase"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomErase</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Compose"><code class="docutils literal notranslate"><span class="pre">Transforms.Compose</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.ToTensor"><code class="docutils literal notranslate"><span class="pre">Transforms.ToTensor</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.PILToTensor"><code class="docutils literal notranslate"><span class="pre">Transforms.PILToTensor</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.ConvertImageDtype"><code class="docutils literal notranslate"><span class="pre">Transforms.ConvertImageDtype</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.ToPILImage"><code class="docutils literal notranslate"><span class="pre">Transforms.ToPILImage</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Normalize"><code class="docutils literal notranslate"><span class="pre">Transforms.Normalize</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Resize"><code class="docutils literal notranslate"><span class="pre">Transforms.Resize</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.CenterCrop"><code class="docutils literal notranslate"><span class="pre">Transforms.CenterCrop</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Pad"><code class="docutils literal notranslate"><span class="pre">Transforms.Pad</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Lambda"><code class="docutils literal notranslate"><span class="pre">Transforms.Lambda</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomApply"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomApply</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomChoice"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomChoice</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomOrder"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomOrder</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomCrop"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomCrop</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomHorizontalFlip"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomHorizontalFlip</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomVerticalFlip"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomVerticalFlip</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomResizedCrop"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomResizedCrop</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.FiveCrop"><code class="docutils literal notranslate"><span class="pre">Transforms.FiveCrop</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.TenCrop"><code class="docutils literal notranslate"><span class="pre">Transforms.TenCrop</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.LinearTransformation"><code class="docutils literal notranslate"><span class="pre">Transforms.LinearTransformation</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.ColorJitter"><code class="docutils literal notranslate"><span class="pre">Transforms.ColorJitter</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomRotation"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomRotation</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomAffine"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomAffine</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.Grayscale"><code class="docutils literal notranslate"><span class="pre">Transforms.Grayscale</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomGrayscale"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomGrayscale</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomPerspective"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomPerspective</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomErasing"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomErasing</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.GaussianBlur"><code class="docutils literal notranslate"><span class="pre">Transforms.GaussianBlur</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.InterpolationMode"><code class="docutils literal notranslate"><span class="pre">Transforms.InterpolationMode</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomInvert"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomInvert</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomPosterize"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomPosterize</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomSolarize"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomSolarize</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomAdjustSharpness"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomAdjustSharpness</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomAutocontrast"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomAutocontrast</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.Transforms.RandomEqualize"><code class="docutils literal notranslate"><span class="pre">Transforms.RandomEqualize</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionMosaic"><code class="docutils literal notranslate"><span class="pre">DetectionMosaic</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionMosaic.input_dim"><code class="docutils literal notranslate"><span class="pre">DetectionMosaic.input_dim</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionMosaic.prob"><code class="docutils literal notranslate"><span class="pre">DetectionMosaic.prob</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionMosaic.enable_mosaic"><code class="docutils literal notranslate"><span class="pre">DetectionMosaic.enable_mosaic</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionMosaic.close"><code class="docutils literal notranslate"><span class="pre">DetectionMosaic.close()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.target_size"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.target_size</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.degrees"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.degrees</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.translate"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.translate</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.scales"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.scales</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.shear"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.shear</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.enable"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.enable</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.filter_box_candidates"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.filter_box_candidates</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.wh_thr"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.wh_thr</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.ar_thr"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.ar_thr</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.area_thr"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.area_thr</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionRandomAffine.close"><code class="docutils literal notranslate"><span class="pre">DetectionRandomAffine.close()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionHSV"><code class="docutils literal notranslate"><span class="pre">DetectionHSV</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionPaddedRescale"><code class="docutils literal notranslate"><span class="pre">DetectionPaddedRescale</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionPaddedRescale.input_dim"><code class="docutils literal notranslate"><span class="pre">DetectionPaddedRescale.input_dim</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionPaddedRescale.swap"><code class="docutils literal notranslate"><span class="pre">DetectionPaddedRescale.swap</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionTargetsFormatTransform"><code class="docutils literal notranslate"><span class="pre">DetectionTargetsFormatTransform</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionTargetsFormatTransform.output_format"><code class="docutils literal notranslate"><span class="pre">DetectionTargetsFormatTransform.output_format</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionTargetsFormatTransform.min_bbox_edge_size"><code class="docutils literal notranslate"><span class="pre">DetectionTargetsFormatTransform.min_bbox_edge_size</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.transforms.DetectionTargetsFormatTransform.max_targets"><code class="docutils literal notranslate"><span class="pre">DetectionTargetsFormatTransform.max_targets</span></code></a></li> </ul> </li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-super_gradients.training.utils">super_gradients.training.utils module</a><ul> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.Timer"><code class="docutils literal notranslate"><span class="pre">Timer</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.Timer.start"><code class="docutils literal notranslate"><span class="pre">Timer.start()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.Timer.stop"><code class="docutils literal notranslate"><span class="pre">Timer.stop()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.HpmStruct"><code class="docutils literal notranslate"><span class="pre">HpmStruct</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.HpmStruct.set_schema"><code class="docutils literal notranslate"><span class="pre">HpmStruct.set_schema()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.HpmStruct.override"><code class="docutils literal notranslate"><span class="pre">HpmStruct.override()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.HpmStruct.to_dict"><code class="docutils literal notranslate"><span class="pre">HpmStruct.to_dict()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.HpmStruct.validate"><code class="docutils literal notranslate"><span class="pre">HpmStruct.validate()</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.WrappedModel"><code class="docutils literal notranslate"><span class="pre">WrappedModel</span></code></a><ul> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.WrappedModel.forward"><code class="docutils literal notranslate"><span class="pre">WrappedModel.forward()</span></code></a></li> <li class="toctree-l4"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.WrappedModel.training"><code class="docutils literal notranslate"><span class="pre">WrappedModel.training</span></code></a></li> </ul> </li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.convert_to_tensor"><code class="docutils literal notranslate"><span class="pre">convert_to_tensor()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.get_param"><code class="docutils literal notranslate"><span class="pre">get_param()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.tensor_container_to_device"><code class="docutils literal notranslate"><span class="pre">tensor_container_to_device()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.adapt_state_dict_to_fit_model_layer_names"><code class="docutils literal notranslate"><span class="pre">adapt_state_dict_to_fit_model_layer_names()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.raise_informative_runtime_error"><code class="docutils literal notranslate"><span class="pre">raise_informative_runtime_error()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.random_seed"><code class="docutils literal notranslate"><span class="pre">random_seed()</span></code></a></li> <li class="toctree-l3"><a class="reference internal" href="super_gradients.training.html#super_gradients.training.utils.torch_version_is_greater_or_equal"><code class="docutils literal notranslate"><span class="pre">torch_version_is_greater_or_equal()</span></code></a></li> </ul> </li> <li class="toctree-l2"><a class="reference internal" href="super_gradients.training.html#module-contents">Module contents</a></li> </ul> </li> </ul> </div> </div> </div> </div> <footer> <hr/> <div role="contentinfo"> <p>© Copyright 2021, SuperGradients team.</p> </div> Built with <a href="https://www.sphinx-doc.org/">Sphinx</a> using a <a href="https://github.com/readthedocs/sphinx_rtd_theme">theme</a> provided by <a href="https://readthedocs.org">Read the Docs</a>. </footer> </div> </div> </section> </div> <script> jQuery(function () { SphinxRtdTheme.Navigation.enable(true); }); </script> </body> </html>
Press p or to see the previous file or, n or to see the next file