Commit 199c3383 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Integrating with gridtk [skip ci]

parent 85089584
Pipeline #12498 skipped
......@@ -7,7 +7,7 @@
Train a Neural network using bob.learn.tensorflow
Usage:
train.py [--iterations=<arg> --validation-interval=<arg> --output-dir=<arg> ] <configuration> [grid <jobs>]
train.py [--iterations=<arg> --validation-interval=<arg> --output-dir=<arg> ] <configuration> [grid --n-jobs <jobs> --name <job-name>]
train.py -h | --help
Options:
......@@ -49,10 +49,14 @@ def main():
job_manager = gridtk.sge.JobManagerSGE()
command = dump_commandline()
dependencies = []
total_jobs = []
for i in range(jobs):
job_id = job_manager.submit(command, dependencies=dependencies)
dependencies = [job_id]
total_jobs.append(job_id)
print("Submitted the jobs {0}".format(total_jobs))
return True
config = imp.load_source('config', args['<configuration>'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment