Fixing hopenet

natanielruiz

2017-09-14 dd62d6fa4a85f18a29de009a972f5599b19ec946

 code/test_on_video.py

@@ -60,7 +60,7 @@
    print 'Loading data.'

    transformations = transforms.Compose([transforms.Scale(224),
    transforms.RandomCrop(224), transforms.ToTensor(),
    transforms.CenterCrop(224), transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])

    model.cuda(gpu)