Batch testing and hopenet training complete

natanielruiz

2017-09-13 c495a0f6b13b794bab9f6e3423d5038ce645d816

 code/train_preangles.py

@@ -134,7 +134,7 @@
    criterion = nn.CrossEntropyLoss().cuda()
    reg_criterion = nn.MSELoss().cuda()
    # Regression loss coefficient
    alpha = 0.00
    alpha = args.alpha

    idx_tensor = [idx for idx in xrange(66)]
    idx_tensor = Variable(torch.FloatTensor(idx_tensor)).cuda(gpu)