Làm thế nào để sử dụng công cụ gỡ lỗi tensorflow tfdbg trên tf.estimator trong Tensorflow?

Question

Tôi đang làm việc với Tensorflow phiên bản 1.4 và tôi muốn gỡ lỗi chức năng train() của mình.

Trong liên kết này https://www.tensorflow.org/programmers_guide/debugger#debugging_tf-learn_estimators_and_experiments

có một cách để làm điều đó cho tf.contrib.learn Estimators, nhưng tôi không thể tìm thấy một cách để thích nghi với nó vào (mới trong phiên bản 1.4) tf.estimator.

Đây là những gì tôi đã cố gắng:

from tensorflow.python import debug as tf_debug 

# Create an estimator 
my_estimator = tf.estimator.Estimator(model_fn=model_fn, 
             params=model_params, 
             model_dir='/tb_dir', 
             config=config_estimator) 

# Create a LocalCLIDebugHook and use it as a hook when calling train(). 
hooks = [tf_debug.LocalCLIDebugHook()] 

# Train 
my_estimator.train(input_fn=train_input_fn, steps=10,hooks=hooks) 

Nhưng tôi đang chạy vào lỗi này:

> --------------------------------------------------------------------------- error 
Traceback (most recent call 
> last) <ipython-input-14-71325f3c8f14> in <module>() 
>  7 
>  8 # Train 
> ----> 9 my_estimator.train(input_fn=train_input_fn, steps=10,hooks=hooks) 
> 
[...] 
> 
> /root/anaconda3/lib/python3.6/site-packages/tensorflow/python/debug/cli/curses_ui.py 
> in _screen_launch(self, enable_mouse_on_start) 
>  443 
>  444  curses.noecho() 
> --> 445  curses.cbreak() 
>  446  self._stdscr.keypad(1) 
>  447 
> 
> error: cbreak() returned ERR 

Ai đó có thể chỉ cho tôi đi đúng hướng?

Answer 1

Giá trị mặc định được đặt để hoạt động trong dòng lệnh, nếu bạn sử dụng IDE như Pycharm, giải pháp đơn giản nhất là thay đổi loại giao diện người dùng.

Hãy thử:

hooks = [tf_debug.LocalCLIDebugHook(ui_type="readline")] 

thay vì:

hooks = [tf_debug.LocalCLIDebugHook()]  

Trong trường hợp bạn sử dụng Pycharm, thêm vào các tham số cấu hình --debug