shape invalid

[Ridiculous]

When running fm and cofm models, no matter how you adjust the parameters, there is always an error "RuntimeError: invalid argument 2: size'[1024 x 1024]' is invalid for input with 1024 elements at ..\src\TH\THStorage.c: 41"
My pytorch version is 0.4.0 and the python version is 3.6
The parameters are
python run_item_recommendation.py -rec_test_files valid.dat:test.dat -l2_lambda 1e-5 -negtive_samples 1 -model_type fm -dataset dbbook2014 -batch_size 1024 -embedding_size 100 -learning_rate 0.1 -topn 10 -seed 3 -eval_training_steps 50000 -interval_steps 50000 -early_stopping_steps_to_wait 2500 -optimizer_type Adam