search

buriburisuri / sugartensor

A slim tensorflow wrapper that provides syntactic sugar for tensor variables. This library will be helpful for practical deep learning researchers not beginners.
MIT License
372 stars 63 forks source link

ValueError: Unable to convert message to str #38

Open SeekPoint opened 6 years ago

SeekPoint commented 6 years ago

import os from hyperparams import Hp import sugartensor as tf

def main(): printLineFileFunc() g = Graph(); print("Graph Loaded") tf.sg_train(optim="Adam", lr=0.0001, lr_reset=True, loss=g.reduced_loss, ep_size=g.num_batch, save_dir='asset/train', max_ep=10, early_stop=False)

I got error:

XX.shape = (8173196, 20) YY.shape = (8173196, 20) Graph Loaded Traceback (most recent call last): File "train.py", line 365, in main(); File "train.py", line 353, in main save_dir='asset/train', max_ep=10, early_stop=False) File "/home/yike.yk/.local/lib/python3.6/site-packages/sugartensor/sg_train.py", line 69, in sg_train train_func(*opt) File "/home/yike.yk/.local/lib/python3.6/site-packages/sugartensor/sg_train.py", line 313, in wrapper with sv.managed_session(config=tf.ConfigProto(allow_soft_placement=True)) as sess: File "/usr/local/lib/python3.6/contextlib.py", line 82, in enter return next(self.gen) File "/usr/local/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 960, in managed_session self.stop(close_summary_writer=close_summary_writer) File "/usr/local/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 788, in stop stop_grace_period_secs=self._stop_grace_secs) File "/usr/local/lib/python3.6/site-packages/tensorflow/python/training/coordinator.py", line 386, in join six.reraise(self._exc_info_to_raise) File "/usr/local/lib/python3.6/site-packages/six.py", line 686, in reraise raise value File "/usr/local/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 949, in managed_session start_standard_services=start_standard_services) File "/usr/local/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 707, in prepare_or_wait_for_session self._write_graph() File "/usr/local/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py", line 610, in _write_graph self._logdir, "graph.pbtxt") File "/usr/local/lib/python3.6/site-packages/tensorflow/python/framework/graph_io.py", line 67, in write_graph file_io.atomic_write_string_to_file(path, str(graph_def)) ValueError: Unable to convert message to str