search

reczoo / FuxiCTR

A configurable, tunable, and reproducible library for CTR prediction https://fuxictr.github.io
Apache License 2.0
914 stars 157 forks source link

Add default param streaming to model_config #84

Closed zhujiem closed 6 months ago

zhujiem commented 6 months ago

streaming: False should be used in dataset_config

zhujiem commented 6 months ago

Fix issue in loading pretrained_emb in npz format in tokenzier.py

      elif pretrain_path.endswith(".npz"):
            npz = np.load(pretrain_path)
            keys = npz["key"]