From 54818d253649ff588ed0054d10dabb2a3a170309 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期四, 10 八月 2017 04:08:12 +0800
Subject: [PATCH] Doing pretty well now with resnet50 and adam with low learning rate. Also fixed test script to use large batches.
---
code/test_on_video.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/code/test_on_video.py b/code/test_on_video.py
index 247c2db..20dfaac 100644
--- a/code/test_on_video.py
+++ b/code/test_on_video.py
@@ -48,9 +48,9 @@
# ResNet101 with 3 outputs.
# model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 23, 3], 66)
# ResNet50
- # model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 6, 3], 66)
+ model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 6, 3], 66)
# ResNet18
- model = hopenet.Hopenet(torchvision.models.resnet.BasicBlock, [2, 2, 2, 2], 66)
+ # model = hopenet.Hopenet(torchvision.models.resnet.BasicBlock, [2, 2, 2, 2], 66)
print 'Loading snapshot.'
# Load snapshot
--
Gitblit v1.8.0