Actually, when I changed the network size smaller, it worked.
Maybe it was just a GPU memory problem?
I still have a question:  Can I make my forward model using forloop like this, not batch tensor format?
Thanks