From 3f99e03cb1a1c37c1da0dd6f98b2715b4890dc1b Mon Sep 17 00:00:00 2001 From: Andre Anjos <andre.anjos@idiap.ch> Date: Tue, 7 Apr 2020 13:13:07 +0200 Subject: [PATCH] [doc/training] Add tips for gridtk --- doc/training.rst | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/doc/training.rst b/doc/training.rst index 65cf6669..2e805e48 100644 --- a/doc/training.rst +++ b/doc/training.rst @@ -16,7 +16,7 @@ To replicate our results, use our main application ``bob binseg train`` followed by the model configuration, and dataset configuration files, and/or command-line options. Use ``bob binseg train --help`` for more information. -.. note:: +.. tip:: We strongly advice training with a GPU (using ``--device="cuda:0"``). Depending on the available GPU memory you might have to adjust your batch @@ -85,6 +85,18 @@ card, for supervised training of baselines. Use it like this: $ bob binseg train -vv <model> ./my_drive_remix.py --batch-size=<see-table> --device="cuda:0" +.. _bob.ip.binseg.gridtk-tip: + +.. tip:: + + If you are at Idiap, you may install the package ``gridtk`` (``conda install + gridtk``) on your environment, and submit the job like this: + + .. code-block:: sh + + $ jman submit --queue=gpu --memory=24G --name=m2unet-drive -- bob binseg train --device='cuda:0' ... #paste the rest of the command-line + + Combined Vessel Dataset (COVD) ============================== -- GitLab