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/train.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/code/train.py b/code/train.py
index e339b10..ff060af 100644
--- a/code/train.py
+++ b/code/train.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