Squeeze function returns a tensor with all the dimensions of input of size 1 removed. It is better to specify the dimension if you want to remove some one dimension, otherwise it may squeeze two or more dimensions. close #209
The parameter of train_epoches will help the model get good performance.
train_epoches
will help the model get good performance.