From 6f0513ea2c5456cf48434489e9647be4582aafd8 Mon Sep 17 00:00:00 2001 From: Brian Bjarke Jensen Date: Tue, 21 May 2024 19:13:24 +0200 Subject: [PATCH] added docs --- model/src/models/visual_communication.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/model/src/models/visual_communication.py b/model/src/models/visual_communication.py index 70c9ae6..4e22778 100644 --- a/model/src/models/visual_communication.py +++ b/model/src/models/visual_communication.py @@ -1,6 +1,7 @@ +"""Definition of VisualCommunicationModel class.""" from __future__ import annotations -import torch.nn as nn +from torch import nn from .angle import AngleTail from .contact import ContactTail @@ -28,6 +29,8 @@ from shared.dto import VisualSyntaxData class VisualCommunicationModel(nn.Module): + """Visual communication model.""" + def __init__(self): super().__init__() # store other models @@ -45,6 +48,7 @@ class VisualCommunicationModel(nn.Module): self.salience_tail = SalienceTail() def forward(self, x): + """Calculate model output on data.""" # generate visual representation vis_rep = self.resnet_head(x) # prepare result map