From e037980d22ce19be70a57863a8610811b6c74047 Mon Sep 17 00:00:00 2001 From: gkaln <55789275+gkaln@users.noreply.github.com> Date: Wed, 28 Apr 2021 14:18:06 +0300 Subject: [PATCH] Update model.py --- model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index 0025284..9b496b6 100644 --- a/model.py +++ b/model.py @@ -1,4 +1,4 @@ -# Andreas Goulas +# Andreas Goulas | first creation import torch import torch.nn as nn @@ -74,4 +74,4 @@ def forward(self, feats, feat_global, device): x = x.view(N, FR, -1) x = torch.cat([x, feat_global], dim=-1) x = self.cls(x) - return x \ No newline at end of file + return x