cd80a074396caa42b81068115c9b066f4ce08414,train_img_model_cent.py,,train,#Any#Any#Any#Any#Any#Any#Any#,193

Before Change

    for batch_idx, (imgs, pids, _) in enumerate(trainloader):
        if use_gpu:
            imgs, pids = imgs.cuda(), pids.cuda()
        imgs, pids = Variable(imgs), Variable(pids)
        outputs, features = model(imgs)
        xentloss = criterion_xent(outputs, pids)
        centloss = criterion_cent(features, pids) * args.weight_cent
        loss = xentloss + centloss

After Change

        for param in criterion_cent.parameters():
            param.grad.data *= (1. / args.weight_cent)
        optimizer_cent.step()
        losses.update(loss.item(), pids.size(0))

        if (batch_idx+1) % args.print_freq == 0:
            print("Batch {}/{}\t Loss {:.6f} ({:.6f})".format(batch_idx+1, len(trainloader), losses.val, losses.avg))

Instances