Thanks. it works. Just wondering, are there some network that has to be trained with double tensor? I keep getting NAN for my network when I use float tensor but it works fine with double tensor.
Is it a sign that my model might have bug?