challenge-ICME2019-Bytedance / Bytedance_ICME_challenge

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

model_zoo/fm.py:56 reshape

codestorm04 opened this issue · comments

Hopefully where will be an error looks like " Input to reshape is a tensor with 100 values, but the requested shape has 200" occurs, when you set a new epoch_size.
By now, the epoch_size = batch_size * field_size = 40 * 5 = 200, coinccidently, so the model_zoo/fm.py:56 57 lines work well, but when you try to predict or the new epoch_size can't be divided by batch_size, the error occurs.