Bihaqo / t3f

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

.value leads to errors #194

Closed faysou closed 4 years ago

faysou commented 4 years ago

When using tensorflow 2 and tensorflow.compat.v1 some functions in "tensor_train_batch.py" and "decompositions.py" have .value after some numbers used for ranks, it leads to the code bugging. When removing these .value the code works again.

For example in tensor_train_batch.py

self._batch_size = tt_cores[0].get_shape()[0].value (line 58)
Bihaqo commented 4 years ago

TF 2 is not fully supported, you have to use

tf.disable_v2_behavior()
tf.enable_eager_execution()

to disable some of v2 features (like in the tutorial).

faysou commented 4 years ago

Ok thank you.