natanielruiz
2017-09-11 653b3608ebe6272510b4c66f445f6f552fdc9ec9
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)