Kissacat

hello, i run your code to train the model. i found my loss including test-loss is always Nan. The output of the model is 0 (dim = 2). "output =...