search

Bihaqo / t3f

Tensor Train decomposition on TensorFlow
https://t3f.readthedocs.io/en/latest/index.html
MIT License
219 stars 55 forks source link

Moving to while loops? #118

Closed Bihaqo closed 6 years ago

Bihaqo commented 6 years ago

Check if using tf.while_loop instead of for loop in loops w.r.t. the TT-cores is faster and reasonably maintainable.

Bihaqo commented 6 years ago

Maybe no need with autograph https://twitter.com/danijarh/status/980223847164100608

Bihaqo commented 6 years ago

Seems that while loops are not necessarily better. https://stackoverflow.com/questions/49714839/tf-map-fn-or-while-loop-for-a-list-of-tensors-of-different-shape/49717913#49717913