search

rixwew / pytorch-fm

Factorization Machine models in PyTorch
MIT License
1.04k stars 225 forks source link

The fc_input_dim seems wrong when "split_half == True" in xdfm.py #12

Closed yxliu-ntu closed 4 years ago

yxliu-ntu commented 4 years ago

when you forward(), the last layer of cin is not split half, while the fc_input_dim still adds cross_layer_size//=2.