From b74b9c54247177c82493f180617d7551de8e2bb1 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期二, 26 九月 2017 03:47:06 +0800
Subject: [PATCH] Before SR experiments
---
code/train_preangles.py | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/code/train_preangles.py b/code/train_preangles.py
index 6622d3f..a2f2027 100644
--- a/code/train_preangles.py
+++ b/code/train_preangles.py
@@ -161,10 +161,10 @@
print 'Ready to train network.'
print 'First phase of training.'
for epoch in range(num_epochs):
- start = time.time()
+ # start = time.time()
for i, (images, labels, cont_labels, name) in enumerate(train_loader):
- print i
- print 'start: ', time.time() - start
+ # print i
+ # print 'start: ', time.time() - start
images = Variable(images).cuda(gpu)
label_yaw = Variable(labels[:,0]).cuda(gpu)
label_pitch = Variable(labels[:,1]).cuda(gpu)
@@ -207,7 +207,7 @@
torch.autograd.backward(loss_seq, grad_seq)
optimizer.step()
- print 'end: ', time.time() - start
+ # print 'end: ', time.time() - start
if (i+1) % 100 == 0:
print ('Epoch [%d/%d], Iter [%d/%d] Losses: Yaw %.4f, Pitch %.4f, Roll %.4f'
--
Gitblit v1.8.0