Skip to content

GitLab

"docs/modules/implicitron/models/visualization/index.rst" did not exist on "322c8f2f50bb95e9d89c47126e081cb57395d116"
Switch branch/tag
Find file BlameHistoryPermalink
  • Da Zheng's avatar
    [Feature] add sparse embedding. (#1497) · cadcc1c2
    Da Zheng authored 
    * add sparse embedding.

* fix

* add test.

* man fixes.

* many fixes

* fix sparse emb.

* fix.

* fix lint.

* fix lint.

* fix kvstore.

* expose DistTensor.

* test sparse embeddings.

* add attach_grad to the backends.

* remove part_id

* fix.

* move backward computation.

* move more computation to backend.

* fix a bug when applying learning rate.

* fix a few things.

* fix a few things.

* add docstring

* fix.

* apply no_grad.

* fix tests.

* fix for other frameworks.

* add examples in docstring.
    cadcc1c2
test_dist_graph_store.py 12.2 KB