From dd62d6fa4a85f18a29de009a972f5599b19ec946 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期四, 14 九月 2017 00:51:53 +0800
Subject: [PATCH] Fixing hopenet

---
 code/train_preangles.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/code/train_preangles.py b/code/train_preangles.py
index 65a2017..3179c24 100644
--- a/code/train_preangles.py
+++ b/code/train_preangles.py
@@ -134,7 +134,7 @@
     criterion = nn.CrossEntropyLoss().cuda()
     reg_criterion = nn.MSELoss().cuda()
     # Regression loss coefficient
-    alpha = 0.00
+    alpha = args.alpha
 
     idx_tensor = [idx for idx in xrange(66)]
     idx_tensor = Variable(torch.FloatTensor(idx_tensor)).cuda(gpu)

--
Gitblit v1.8.0