DataParallel model stuck on loss.backward()

[winchest (ZHAO)]

For someone might have the same problem, U can see this post:

Training on the multi-GPUs but stuck in loss.backward()

OMGGG！Finally I find that the os.environ code part seems to be conflict with torch.nn.DataParallel.So when I delete the os.environ code It worked!!

Hope it will be helpful!