Commit 29525139 authored by Constantin Pape's avatar Constantin Pape

Fix issues with submission script

parent 0026647e
......@@ -10,7 +10,7 @@ setup(
entry_points={
"console_scripts": [
"view_data = tools_impl.view_data:main",
"submit_slurm = tools_impl.submit_to_slurm:main"
"submit_slurm = tools_impl.submit_slurm:main"
]
},
)
......@@ -21,7 +21,7 @@ def write_slurm_template(script, out_path, env_name,
"#SBATCH -t %i\n"
"#SBATCH --qos=%s\n"
"#SBATCH -p gpu\n"
"#SBATCH -C gpu=%s\n"
"#SBATCH -C \"gpu=%s\"\n"
"#SBATCH --gres=gpu:%i\n") % (n_threads,
mem_limit, time_limit,
qos, gpu_type, n_gpus)
......@@ -39,13 +39,13 @@ def write_slurm_template(script, out_path, env_name,
def submit_slurm(script, input_, n_threads=7, n_gpus=1,
gpu_type='2080Ti|1080Ti', mem_limit='64G',
gpu_type='2080Ti', mem_limit='64G',
time_limit=TWO_DAYS, qos='normal',
env_name=None, mail_address=None):
""" Submit python script that needs gpus with given inputs on a slurm node.
"""
tmp_folder = os.path.expanduser('~/.deep-cell/slurm')
tmp_folder = os.path.expanduser('~/.ai4ia/slurm')
os.makedirs(tmp_folder, exist_ok=True)
print("Submitting training script %s to cluster" % script)
......
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