From fe6093e8250e730bb68f3cb4d2cdab6a5ee9b282 Mon Sep 17 00:00:00 2001
From: Lu Li <42978639+luli-git@users.noreply.github.com>
Date: Sat, 4 Nov 2023 00:25:23 -0400
Subject: [PATCH] Update ginet_finetune.py

Line 113 and line 127 are repetitive logic.
---
 models/ginet_finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/ginet_finetune.py b/models/ginet_finetune.py
index 49fc139..91ecc14 100644
--- a/models/ginet_finetune.py
+++ b/models/ginet_finetune.py
@@ -110,7 +110,7 @@ def __init__(self,
                     nn.Linear(self.feat_dim//2, self.feat_dim//2), 
                     nn.ReLU(inplace=True),
                 ])
-            pred_head.append(nn.Linear(self.feat_dim//2, out_dim))
+
         elif pred_act == 'softplus':
             pred_head = [
                 nn.Linear(self.feat_dim, self.feat_dim//2),