Skip to content

tensornet-0.1.0
Compare
Choose a tag to compare
@zhangys-lucky zhangys-lucky released this 11 Sep 06:49
· 147 commits to master since this release
v0.1.0
ea19d0a

TensorNet V-0.1.0

First version of TensorNet.

In this version we have published tensornet with async train mode support which we have tested completely.

the main API contained:

  1. tn.distribute.PsStrategy.
    tn.distribute.PsStrategy with same interface with tensorflow's strategy in order to cluster management.

  2. tn.feature_column.category_column
    tn.feature_column.category_column is one of the important API of tensornet, with which we could define sparse feature column support dimension close to 2**64. tn.feature_column.category_column has the same interface with tf.feature_column.

  3. tn.layers.EmbeddingFeatures
    tn.layers.EmbeddingFeatures is the second important API in tensornet, in which we pull and push sparse embeddiong vector from parameter server.

  4. tn.optimizer.Optimizer
    We wrapped tensorflow optimizer in tn.optimizer.Optimizer, this is mainly used in asyc train mode, in which we intercept tensorflow's gradients update logic, and update gradients in parameter server asynchronously.

  5. tn.model.Model
    We inherited keras.layers.model and override its save model method to support save and load sparse feature in parameter server.

Assets 2
Loading