From beb9f36419d0df03c3248757f54af032a633e05c Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期六, 12 八月 2017 11:49:10 +0800
Subject: [PATCH] AFLW training ready.
---
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