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.