search

DeepRec-AI / HybridBackend

A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster
Apache License 2.0
156 stars 30 forks source link

[DIST] Multiple enhancement in `hb.keras`. #91

Closed francktcheng closed 1 year ago

francktcheng commented 1 year ago
  1. Support printing info when loading weights in Keras. a. the ratio of filtered variables by the scope arguments of model.load_weights is print. b. the ratio of loading variables that hits the ckpt is print. #86
  2. Support the usage of multiple labels and nested metrics in hb.keras. #85 #90
  3. Support printing layer's info in Keras. #87