From c495a0f6b13b794bab9f6e3423d5038ce645d816 Mon Sep 17 00:00:00 2001 From: natanielruiz <nataniel777@hotmail.com> Date: 星期三, 13 九月 2017 21:12:59 +0800 Subject: [PATCH] Batch testing and hopenet training complete --- code/test_preangles.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/code/test_preangles.py b/code/test_preangles.py index 67e4744..4aedfd8 100644 --- a/code/test_preangles.py +++ b/code/test_preangles.py @@ -43,7 +43,7 @@ cudnn.enabled = True gpu = args.gpu_id - snapshot_path = os.path.join('output/snapshots', args.snapshot + '.pkl') + snapshot_path = args.snapshot # ResNet101 with 3 outputs. # model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 23, 3], 66) -- Gitblit v1.8.0