From abe876183052e9da9c3d633e41386c5c1f4fc1e6 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期五, 08 九月 2017 05:51:42 +0800
Subject: [PATCH] Before adding refinement layer
---
code/train.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/code/train.py b/code/train.py
index d80ed30..5d7fc7d 100644
--- a/code/train.py
+++ b/code/train.py
@@ -190,7 +190,7 @@
if epoch % 1 == 0 and epoch < num_epochs - 1:
print 'Taking snapshot...'
torch.save(model.state_dict(),
- 'output/snapshots/resnet50_norm_norot_epoch_'+ str(epoch+1) + '.pkl')
+ 'output/snapshots/resnet50_norm_30rot_epoch_'+ str(epoch+1) + '.pkl')
# Save the final Trained Model
- torch.save(model.state_dict(), 'output/snapshots/resnet50_norm_norot_epoch_' + str(epoch+1) + '.pkl')
+ torch.save(model.state_dict(), 'output/snapshots/resnet50_norm_30rot_epoch_' + str(epoch+1) + '.pkl')
--
Gitblit v1.8.0