dlc2action.model

 1#
 2# Copyright 2020-present 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. 
 5# A copy is included in dlc2action/LICENSE.AGPL.
 6#
 7"""
 8## Models
 9
10The `dlc2action.model.base_model.Model` abstract class inherits from `torch.nn.Module` but additionally
11handles automatic integration
12of SSL modules (see `dlc2action.ssl`) and enforces consistent input and output formats.
13"""
14
15from dlc2action.model.ms_tcn import *
16
17__pdoc__ = {
18    "ms_tcn.MS_TCN3.dump_patches": False,
19    "ms_tcn.MS_TCN3.training": False,
20    "ms_tcn_modules": False,
21    "base_model.Model.dump_patches": False,
22    "base_model.Model.training": False,
23    "base_model.LoadedModel.dump_patches": False,
24    "base_model.LoadedModel.training": False,
25}