cfd0fea779274f79a49bba750b113579f16e962c,train_vidreid_xent_htri.py,,main,#,108

Before Change

            model.load_state_dict(model_dict)
            print("Loaded pretrained weights from "{}"".format(args.load_weights))

    if args.resume:
        if check_isfile(args.resume):
            checkpoint = torch.load(args.resume)
            model.load_state_dict(checkpoint["state_dict"])
            args.start_epoch = checkpoint["epoch"]
            rank1 = checkpoint["rank1"]
            print("Loaded checkpoint from "{}"".format(args.resume))
            print("- start_epoch: {}\n- rank1: {}".format(args.start_epoch, rank1))

    if use_gpu:
        model = nn.DataParallel(model).cuda()

    if args.evaluate:

After Change

        model.load_state_dict(model_dict)
        print("Loaded pretrained weights from "{}"".format(args.load_weights))

    if args.resume and check_isfile(args.resume):
        checkpoint = torch.load(args.resume)
        model.load_state_dict(checkpoint["state_dict"])
        args.start_epoch = checkpoint["epoch"] + 1
        best_rank1 = checkpoint["rank1"]
        print("Loaded checkpoint from "{}"".format(args.resume))
        print("- start_epoch: {}\n- rank1: {}".format(args.start_epoch, best_rank1))

    if use_gpu:
        model = nn.DataParallel(model).cuda()

    if args.evaluate:

Instances