dlc2action.loss

Losses

There is no dedicated loss class in dlc2action. Instead we use regular torch.nn.Module instances that take prediction and target as input and return loss value as output.

View Source

 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## Losses
 8
 9There is no dedicated loss class in `dlc2action`. Instead we use regular `torch.nn.Module` instances that take
10prediction
11and target as input and return loss value as output.
12"""
13
14from dlc2action.loss.contrastive import *
15from dlc2action.loss.mse import *
16from dlc2action.loss.ms_tcn import *
17
18__pdoc__ = {
19    "contrastive.NTXent.dump_patches": False,
20    "contrastive.NTXent.training": False,
21    "contrastive.TripletLoss.dump_patches": False,
22    "contrastive.TripletLoss.training": False,
23    "contrastive.CircleLoss.dump_patches": False,
24    "contrastive.CircleLoss.training": False,
25    "ms_tcn.MS_TCN_Loss.dump_patches": False,
26    "ms_tcn.MS_TCN_Loss.training": False,
27    "mse.MSE.dump_patches": False,
28    "mse.MSE.training": False,
29    "asymmetric_loss": False,
30}