fix: Remove race condition during multi-GPU training#39

uermel wants to merge 2 commits into

MLI-lab:masterMLI-lab/DeepDeWedge:masterfrom

uermel:masteruermel/DeepDeWedge:masterCopy head branch name to clipboard