Skip to content

Commit 40ebcb1

Browse files
committed
block multi-GPU processing with PyTorch 1.2+ to avoid [pytorch/pytorch#21108]
1 parent df4e9c0 commit 40ebcb1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

espnet/bin/asr_train.py

+1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515

1616
import configargparse
1717
import numpy as np
18+
import torch
1819

1920
from espnet.utils.cli_utils import strtobool
2021
from espnet.utils.training.batchfy import BATCH_COUNT_CHOICES

0 commit comments

Comments
 (0)