I have sloived the problem in my case.
my len(train_data) = 55937 and my batchsize = 64 >> 1, It looks like no problem.
but I have found that 55937 % 64  = 1, which means the last batchsize =1,
so runing_var becomes nan after 1 epoch.
hope it helps you.