From 898871a1ca98e6253d7aaaa7a47fac4bfd8f6833 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期一, 11 九月 2017 09:09:54 +0800
Subject: [PATCH] Testing regression loss
---
code/hopenet.py | 6 ------
1 files changed, 0 insertions(+), 6 deletions(-)
diff --git a/code/hopenet.py b/code/hopenet.py
index 274044f..5bac804 100644
--- a/code/hopenet.py
+++ b/code/hopenet.py
@@ -88,12 +88,6 @@
return nn.Sequential(*layers)
- def get_expectation(angle):
- angle_pred = F.softmax(angle)
-
- angle_pred = torch.sum(angle_pred.data * self.idx_tensor, 1)
- return angle_pred
-
def forward(self, x):
x = self.conv1(x)
x = self.bn1(x)
--
Gitblit v1.8.0