Skip to content

Commit

Permalink
fix: DataLoader freeze
Browse files Browse the repository at this point in the history
  • Loading branch information
tueda committed Aug 13, 2024
1 parent ed0824d commit 571b22f
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions .github/workflows/scripts/run-tutorial-notebook.sh
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,12 @@ time {
echo "::endgroup::"

cd "$working_directory"

# Prevent DataLoader from freezing during training.
# https://github.com/pytorch/pytorch/issues/15808#issuecomment-1291514752
export OMP_NUM_THREADS=1
export MKL_NUM_THREADS=1

# We may set a timeout (for each cell) to ensure in-progress notebooks and logs are saved,
# even if the workflow hits the total 6-hour limit in GitHub Actions.
if [ "$timeout" -ne 0 ]; then
Expand Down

0 comments on commit 571b22f

Please sign in to comment.