From e306e86e925e4211c1c2d2f68de45d5e55f3e215 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期日, 13 八月 2017 01:22:31 +0800
Subject: [PATCH] Some cleanup and AFLW working
---
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