def submitNoiseSweep(p, gEp, gIp, noisep, ENV, simRootDir, simLabel, appName, rtLimit, numCPU, blocking, timePrefix, numRepeat, dry_run, extraIterparams={}): ''' Submit a parameter sweep with an extra dimension: noise_sigma. ''' noise_sigma_arr = [] g_AMPA_total_arr = [] g_GABA_total_arr = [] for n in noisep.values(): for E_coupling in gEp.values(): for I_coupling in gIp.values(): noise_sigma_arr.append(n) g_AMPA_total_arr.append(E_coupling) g_GABA_total_arr.append(I_coupling) iterparams = { 'noise_sigma': np.array(noise_sigma_arr), 'g_AMPA_total': np.array(g_AMPA_total_arr), 'g_GABA_total': np.array(g_GABA_total_arr), #'noise_sigma' : [150], #'g_AMPA_total' : [1020], #'g_GABA_total' : [3060] } iterparams.update(extraIterparams) ac = ArgumentCreator(p, printout=True) ac.insertDict(iterparams, mult=False) ############################################################################### submitter = SubmitterFactory.getSubmitter(ac, appName, envType=ENV, rtLimit=rtLimit, output_dir=simRootDir, label=simLabel, blocking=blocking, timePrefix=timePrefix, numCPU=numCPU) ac.setOption('output_dir', submitter.outputDir()) startJobNum = 0 submitter.submitAll(startJobNum, numRepeat, dry_run=dry_run) submitter.saveIterParams(iterparams, dry_run=dry_run)
simRootDir = 'output_local' simLabel = 'test spaces' parameters = {} startJobNum = 10 numRepeat = 1 # Submitting ac = ArgumentCreator(parameters) appName = 'dummy_script.py' rtLimit = '00:05:00' blocking = True timePrefix = False submitter = SubmitterFactory.getSubmitter(ac, appName, envType=ENV, rtLimit=rtLimit, output_dir=simRootDir, label=simLabel, blocking=blocking, timePrefix=timePrefix); ac.setOption('output_dir', submitter.outputDir()) iterparams = { 'output_data' : np.arange(10) } ac.insertDict(iterparams, mult=False) submitter.submitAll(startJobNum, numRepeat, dry_run=False)
dirs = ('{0}pA' , (31, 31)) for noise_sigma in noise_sigma_all: p = {} simLabel = dirs[0].format(int(noise_sigma)) rowN = dirs[1][0] colN = dirs[1][1] p['shapeRows'] = rowN p['shapeCols'] = colN ############################################################################### ac = ArgumentCreator(p, printout=True) iterparams = { 'row' : np.arange(rowN), 'col' : np.arange(colN) #'row' : np.arange(1), #'col' : np.arange(1) } ac.insertDict(iterparams, mult=True) ############################################################################### submitter = SubmitterFactory.getSubmitter(ac, appName, envType=ENV, rtLimit=rtLimit, output_dir=simRootDir, label=simLabel, blocking=blocking, timePrefix=timePrefix, numCPU=numCPU) ac.setOption('output_dir', submitter.outputDir()) startJobNum = 0 submitter.submitAll(startJobNum, numRepeat, dry_run=dry_run)