trainer.py

[FMfelix]

Dear Professor,
There is an error on line 674 of the trainer.py: tmp_tensor = tf.compat.v1.get_default_graph().get_tensor_by_name(k)
When I debug, I find that the shape of tmp_tensor is empty（），it makes the whole pre-training value unable to be loaded.
I am looking forward to your reply as soon as possible

[cyberfag]

I ran into similar problem and it turned out to be due to versions issue. Make sure you use tensorflow-gpu 1.x and Python 3.6.x (the authors mentioned 3.5 but it didn't work for me as well as initial 3.8)