From ec99c6649af6bdbd3c836f20cdc81170e7045cc8 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期四, 14 九月 2017 10:06:48 +0800
Subject: [PATCH] Training hopenet and normal for different alpha values on AFLW
---
code/train.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/code/train.py b/code/train.py
index 03d5cf5..fd0735a 100644
--- a/code/train.py
+++ b/code/train.py
@@ -251,7 +251,7 @@
loss_pitch += alpha * loss_reg_pitch
loss_roll += alpha * loss_reg_roll
- loss_yaw *= 0.35
+ loss_yaw *= 1
# Finetuning loss
loss_seq = [loss_yaw, loss_pitch, loss_roll]
--
Gitblit v1.8.0