Skip to content

Commit 95a21be

Browse files
committed
update run rwmh nf script
1 parent 1ee8c45 commit 95a21be

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

example/real_data_expt/rwmh/run_rwmh.nf

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4,19 +4,19 @@ include { combine_csvs; } from '../../nf-nest/combine.nf'
44

55
params.dryRun = false
66
params.n_sample = params.dryRun ? 8 : 64
7-
params.nrunThreads = 1
7+
params.nrunThreads = 4
88

99
def julia_env = file("${moduleDir}/../../julia_env")
1010
def julia_script = file(moduleDir/'run_rwmh.jl')
1111
// def plot_script = file(moduleDir/'tuning.jl')
1212

1313
def variables = [
1414
seed: 1..32,
15-
target: ["TReg", "Brownian", "Sonar", "SparseRegression", "LGCP"],
16-
flowtype: ["BackwardIRFMixFlow", "DeterministicMixFlow", "EnsembleIRFFlow", "IRFMixFlow"],
15+
target: ["Sonar", "Brownian", "TReg", "SparseRegression"],
16+
flowtype: ["IRFMixFlow"],
1717
kernel: ["MF.RWMH"],
1818
nchains: [30],
19-
flow_length: [5000],
19+
flow_length: [3000],
2020
]
2121

2222
workflow {
@@ -30,17 +30,17 @@ workflow {
3030

3131
process run_simulation {
3232
debug false
33-
memory { 4.GB * Math.pow(2, task.attempt-1) }
34-
time { 24.hour * Math.pow(2, task.attempt-1) }
35-
cpus 1
36-
// errorStrategy { task.attempt < 2 ? 'retry' : 'ignore' }
33+
memory { 10.GB * Math.pow(2, task.attempt-1) }
34+
time { 2.hour * Math.pow(2, task.attempt-1) }
35+
cpus 4
36+
errorStrategy { task.attempt < 3 ? 'retry' : 'ignore' }
3737
input:
3838
path julia_env
3939
val config
4040
output:
4141
path "${filed(config)}"
4242
"""
43-
${activate(julia_env,params.nrunThreads)}
43+
${activate(julia_env,10)}
4444
4545
include("$julia_script")
4646
@@ -53,7 +53,7 @@ process run_simulation {
5353
nchains = ${config.nchains}
5454
5555
# run simulation
56-
df, output = run_simulation(seed, name, flowtype, kernel, T, nchains; nsample = ${params.n_sample})
56+
df, _ = run_simulation(seed, name, flowtype, kernel, T, nchains; nsample = ${params.n_sample})
5757
5858
# store output
5959
mkdir("${filed(config)}")

0 commit comments

Comments
 (0)