diff --git a/tutorials/torch_example.py b/tutorials/torch_example.py index e671a52a..3556b12e 100755 --- a/tutorials/torch_example.py +++ b/tutorials/torch_example.py @@ -3,6 +3,7 @@ This script was tested on a single compute node with 4 GPUs. Note that you need to adapt ``GPUS_PER_NODE`` (see ll. 25). """ + import logging import pathlib import random