Skip to content
Snippets Groups Projects
Commit 199c3383 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Integrating with gridtk [skip ci]

parent 85089584
Branches
Tags
1 merge request!14gridtk integration
Pipeline #
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
Train a Neural network using bob.learn.tensorflow Train a Neural network using bob.learn.tensorflow
Usage: 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 train.py -h | --help
Options: Options:
...@@ -49,10 +49,14 @@ def main(): ...@@ -49,10 +49,14 @@ def main():
job_manager = gridtk.sge.JobManagerSGE() job_manager = gridtk.sge.JobManagerSGE()
command = dump_commandline() command = dump_commandline()
dependencies = [] dependencies = []
total_jobs = []
for i in range(jobs): for i in range(jobs):
job_id = job_manager.submit(command, dependencies=dependencies) job_id = job_manager.submit(command, dependencies=dependencies)
dependencies = [job_id] 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>']) config = imp.load_source('config', args['<configuration>'])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment