From 8e5d7dbfe49d194b7d0b616307663e9a88fbcd88 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期五, 15 九月 2017 06:04:49 +0800
Subject: [PATCH] Training AFLW
---
code/train_preangles.py | 2 --
1 files changed, 0 insertions(+), 2 deletions(-)
diff --git a/code/train_preangles.py b/code/train_preangles.py
index 6328ef2..31144f6 100644
--- a/code/train_preangles.py
+++ b/code/train_preangles.py
@@ -197,8 +197,6 @@
loss_pitch += alpha * loss_reg_pitch
loss_roll += alpha * loss_reg_roll
- loss_yaw *= 1
-
loss_seq = [loss_yaw, loss_pitch, loss_roll]
# loss_seq = [loss_reg_yaw, loss_reg_pitch, loss_reg_roll]
grad_seq = [torch.Tensor(1).cuda(gpu) for _ in range(len(loss_seq))]
--
Gitblit v1.8.0