dlc2action.model

Models

The dlc2action.model.base_model.Model abstract class inherits from torch.nn.Module but additionally handles automatic integration of SSL modules (see dlc2action.ssl) and enforces consistent input and output formats.

 1#
 2# Copyright 2020-2022 by A. Mathis Group and contributors. All rights reserved.
 3#
 4# This project and all its files are licensed under GNU AGPLv3 or later version. A copy is included in dlc2action/LICENSE.AGPL.
 5#
 6"""
 7## Models
 8
 9The `dlc2action.model.base_model.Model` abstract class inherits from `torch.nn.Module` but additionally
10handles automatic integration
11of SSL modules (see `dlc2action.ssl`) and enforces consistent input and output formats.
12"""
13
14__pdoc__ = {
15    "ms_tcn.MS_TCN3.dump_patches": False,
16    "ms_tcn.MS_TCN3.training": False,
17    "ms_tcn_modules": False,
18    "base_model.Model.dump_patches": False,
19    "base_model.Model.training": False,
20    "base_model.LoadedModel.dump_patches": False,
21    "base_model.LoadedModel.training": False,
22}