search

datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
MIT License
392 stars 73 forks source link

update the embedding initialization method and fix bugs #17

Closed yinpu closed 2 years ago

yinpu commented 2 years ago

update the embedding initialization method and fix bugs Embedding initializers need the embedding size and vocab size, so add get_embedding_layer() method in feature class