search

rixwew / pytorch-fm

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

some bug？ #3

Closed AmazingDD closed 5 years ago

AmazingDD commented 5 years ago

1.In the forward process, the dimension of input X in all models are (batch_size, field_size), here can I treat all these field size as feature size?

  1. in fnfm.py, you forgot define self.embed_layer in forward function methed. please check.