search

meta-soul / MetaSpore

A unified end-to-end machine intelligence platform
Apache License 2.0
642 stars 118 forks source link

Specify the dimension of squeeze function in din model #208

Closed Cheng-Su closed 2 years ago

Cheng-Su commented 2 years ago
  1. 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
  2. The parameter of train_epoches will help the model get good performance.