run_SA_EA.py

import argparse
import os
import numpy as np
from models import EdwardsAnderson
from python_interface import ClassicalAnneal

if __name__ == "__main__":

    if not os.path.exists('./results/'):
        os.mkdir('./results')
    if not os.path.exists('./results/EA'):
        os.mkdir('./results/EA')
    if not os.path.exists('./results/EA/SA'):
        os.mkdir('./results/EA/SA')

    # CHANGE THE DEFAULT ARGUMENTS HERE

    parser = argparse.ArgumentParser()

    parser.add_argument('--seed', default=1)
    parser.add_argument('--tau_schedule', default=[2**i for i in range(1,13+1)])
    parser.add_argument('--mcsteps', default=5) #Number of sweeps
    parser.add_argument('--num_warmup', type = int, default=2000) #Number of warmup steps
    parser.add_argument('--numruns', default=25, type=int)

    args = parser.parse_args()
    realization = args.seed
    num_warmup = args.num_warmup
    nrows = 40
    ncols = 40
    numruns = args.numruns

    gs_fname = './data/'+str(nrows)+'x'+str(ncols)+'/gs_seed'+str(realization)+'.txt'
    interactions_fname = './data/EA_'+str(nrows)+'x'+str(ncols)+'/'+str(nrows)+'x'+str(ncols)+'_uniform_seed'+str(realization)+'.txt'

    model = EdwardsAnderson(nrows=nrows, ncols=ncols, gs_fname=gs_fname, interactions_fname=interactions_fname)

    Energies = np.zeros((numruns, len(args.tau_schedule)), np.float64)
    checkpointfile = './results/EA/SA/EA_'+str(nrows)+'x'+str(ncols)+'_SA_realization'+str(realization)+'_numwarmup'+str(num_warmup)+'_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 number ", annealingrun)
        SA = ClassicalAnneal(model, latticetype = "2D", annealingrunseed = annealingrun, **vars(args))
        Energies[annealingrun-1] = SA.perform_tau_schedule()
        np.save(checkpointfile, Energies[:annealingrun])