run_PIQMC_SK.py

import argparse
import os
import numpy as np
from models import SK
from python_interface import QuantumPIAnneal

if __name__ == "__main__":

    if not os.path.exists('./results/'):
        os.mkdir('./results')
    if not os.path.exists('./results/SK'):
        os.mkdir('./results/SK')
    if not os.path.exists('./results/SK/PIQMC'):
        os.mkdir('./results/SK/PIQMC')

    parser = argparse.ArgumentParser()

    parser.add_argument('--seed', default=1)
    parser.add_argument('--P', default=100)
    parser.add_argument('--gamma_0', default=2.0, type=float)
    parser.add_argument('--tau_schedule', default=[2**i for i in range(1,14+1)])
    parser.add_argument('--mcsteps', default=5) #Number of sweeps
    parser.add_argument('--numruns', default=50, type=int)

    args = parser.parse_args()
    realization = args.seed
    P = args.P
    numruns = args.numruns

    N = 100

    interactions_fname = './data/SK_N'+str(N)+'/'+str(N)+'_SK_seed'+str(realization)+'.txt'
    model = SK(nspins=N, interactions_fname=interactions_fname)

    Energies = np.zeros((numruns, len(args.tau_schedule),int(P)), np.float64)
    checkpointfile = './results/SK/PIQMC/SK_N'+str(N)+'_PIQMC_realization'+str(realization)+'_Energies.npy'
    try:
        print("Loading checkpoint!")
        Loaded = np.load(checkpointfile)
        Energies[:Loaded.shape[0]] = Loaded
    except:
        print("Failed! Running from scratch")
        Loaded = []

    for annealingrun in range(len(Loaded)+1, numruns+1):
        print("annealing run = ", annealingrun)
        Q = QuantumPIAnneal(model, latticetype = "FullyConnected",  **vars(args))
        Energies[annealingrun-1] = Q.perform_tau_schedule()
        np.save(checkpointfile, Energies[:annealingrun])