2733bef356c53286d475a67476d88d4840923830,code/deep/finetune_AlexNet_ResNet/finetune_office31.py,,finetune,#,87

Before Change

            if phase == "tar" and epoch_acc > best_acc:
                best_acc = epoch_acc
        print()
        fname = "finetune_result" + model_name + \
            str(LEARNING_RATE) + str(args.source) + \
            "-" + str(args.target) + ".csv"
        np.savetxt(fname, np.asarray(a=acc_hist, dtype=float), delimiter=",",
                   fmt="%.4f")
    time_pass = time.time() - since
    print("Training complete in {:.0f}m {:.0f}s".format(

After Change

            if phase == "val" and epoch_acc > best_acc:
                stop = 0
                best_acc = epoch_acc
                torch.save(model.state_dict(), "model.pkl")
        if stop >= args.early_stop:
            break
        print()
    model.load_state_dict(torch.load("model.pkl"))

Instances