natanielruiz
2017-09-13 c495a0f6b13b794bab9f6e3423d5038ce645d816
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)