search

datawhalechina / torch-rechub

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

Add feature: Support custom embedding layer initializers #9

Closed yinpu closed 2 years ago

yinpu commented 2 years ago
  1. Support setting the embedding initialization method:RandomNormal、RandomUniform、XavierNormal、XavierUniform
  2. Support loading pre-trained embeddings：Pretrain