search

Qihoo360 / tensornet

Apache License 2.0
315 stars 72 forks source link

forget _apply_sparse_duplicate_indices? #29

Open zhangys-lucky opened 3 years ago

zhangys-lucky commented 3 years ago

tensorflow 在apply sparse gradient的时候先对重复的indice做了去重，tensornet是否遗漏了相关的去重工作。 https://github.com/tensorflow/tensorflow/blob/5681c179eff80bce00e526303950b67b23cad14c/tensorflow/python/keras/optimizer_v2/optimizer_v2.py#L1100