RuntimeError: Function 'SqrtBackward' returned nan values in its 0th output

[zl535320706]

can use
torch.sqrt(x + 1e-8)
replace
torch.sqrt(x)
to solve this problem