Commit 762699dd authored by Fernando Perez-Garcia's avatar Fernando Perez-Garcia

Use custom application to train adn infer

parent 7eac2cdd
......@@ -12,6 +12,8 @@ class Job:
self.requested_time = '23:59:59'
self.tmem = '11.5G'
self.config_path = config_path
self.train = True
self.infer = True
def write(self, old_cluster=False):
......@@ -56,7 +58,13 @@ class Job:
a(' install --user -r requirements-gpu.txt')
# Custom code
for action in ('train', 'inference'):
actions = []
if self.train:
actions.append('train')
if self.infer:
actions.append('inference')
for action in actions:
a('')
a('LD_LIBRARY_PATH=\\')
a('"/share/apps/libc6_2.23/lib/x86_64-linux-gnu\\')
......@@ -70,7 +78,8 @@ class Job:
a(' $(command -v\\')
a(' /share/apps/python-3.6.3-shared/bin/python3) \\')
a(f' -u net_segment.py {action} \\')
a(f' -c {self.config_path}')
a(f' -c "{self.config_path}" \\')
a(f' -a vesseg_applications.segmentation_application_vessels.SegmentationApplicationVessels')
a('')
if old_cluster:
a('fi')
......
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