From d285712b1c0a3e0785a4e5b17debdde650a7c26a Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期三, 13 九月 2017 21:34:38 +0800
Subject: [PATCH] Removed hourglass

---
 code/batch_testing_preangles.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/code/batch_testing_preangles.py b/code/batch_testing_preangles.py
index 11390b0..bf0d32b 100644
--- a/code/batch_testing_preangles.py
+++ b/code/batch_testing_preangles.py
@@ -50,7 +50,7 @@
     # 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, 0)
     # ResNet18
     # model = hopenet.Hopenet(torchvision.models.resnet.BasicBlock, [2, 2, 2, 2], 66)
 

--
Gitblit v1.8.0