Hi, Branch py3 working fine on my PC, I use ubuntu18.04, py3.5 and tensorflow 1.10

session run is gray in tensorboard->graph, and unknown device about stock-rnn HOT 2 CLOSED

lilianweng commented on May 20, 2024

session run is gray in tensorboard->graph, and unknown device

from stock-rnn.

Comments (2)

elulue commented on May 20, 2024

btw, if use tfdbg, I can see -
"WARNING:tensorflow:Failed to load partition graphs for device /job:localhost/replica:0/task:0/device:CPU:0 from disk. As a fallback, the client graphs will be used. This may cause mismatches in device names."

from stock-rnn.

elulue commented on May 20, 2024

I've found the root cause, tensorboard don't record compute time etc by default.
I updated below and working for me -


                if np.mod(global_step, show_every_n_step) == 1:
                    run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
                    run_metadata = tf.RunMetadata()

                    train_loss, train_mse, _, train_merged_sum = self.sess.run(
                        [self.loss, self.mse, self.optim, self.summary], train_data_feed,
                        options=run_options, run_metadata=run_metadata)

                    self.writer.add_run_metadata(run_metadata, 'step{}'.format(global_step))
                    self.writer.add_summary(train_merged_sum, global_step=global_step)

                else:
                    train_loss, train_mse, _, train_merged_sum = self.sess.run(
                        [self.loss, self.mse, self.optim, self.summary], train_data_feed)
                    self.writer.add_summary(train_merged_sum, global_step=global_step)

from stock-rnn.

Recommend Projects