Avoid NaN errors and misclassifications in tests
Avoid NaN errors and misclassifications in tests

ethantang95/DIGITS,ethantang95/DIGITS-GAN,gheinrich/DIGITS-GAN,TimZaman/DIGITS,ethantang95/DIGITS,TimZaman/DIGITS,Deepomatic/DIGITS,TimZaman/DIGITS,ethantang95/DIGITS,gheinrich/DIGITS-GAN,ethantang95/DIGITS,gheinrich/DIGITS-GAN,bygreencn/DIGITS,ethantang95/DIGITS-GAN,bygreencn/DIGITS,ethantang95/DIGITS-GAN,ethantang95/DIGITS-GAN,gheinrich/DIGITS-GAN,Deepomatic/DIGITS,TimZaman/DIGITS,bygreencn/DIGITS,bygreencn/DIGITS,Deepomatic/DIGITS,Deepomatic/DIGITS