Number of threads explodes when training

[nikwoj]

In the training script, an additional environment variable needs to be set in the worker_init_fn function

os.environ['OMP_NUM_THREADS'] = 1

See torch comment for additional information

[rasmushaugaard]

Thank you, I'll add that :)