Tôi đang sử dụng Keras để đào tạo CNN bằng chức năng fit_generator.Phân phối và biểu đồ TensorBoard với Keras và fit_generator
Có vẻ như là known issue rằng TensorBoard không hiển thị biểu đồ và bản phân phối trong thiết lập này.
Có ai tìm ra cách để nó hoạt động không?
Không có cách dễ dàng để chỉ cần cắm nó với một dòng mã, bạn phải viết tóm tắt của bạn bằng tay.
Tin tốt là nó không phải là khó khăn và bạn có thể sử dụng mã callback TensorBoard trong Keras như một tài liệu tham khảo: https://github.com/fchollet/keras/blob/master/keras/callbacks.py#L537
Về cơ bản, hãy viết một hàm ví dụ write_summaries(model) và gọi nó là bất cứ khi nào bạn muốn viết tóm tắt của bạn (ví dụ chỉ sau fit_generator() của bạn)
Bên write_summaries(model) sử dụng chức năng của bạn tf.summary, histogram_summary và các chức năng tóm tắt để đăng nhập dữ liệu bạn muốn xem trên tensorboard.
Nếu bạn không biết chính xác như thế nào, kiểm tra hướng dẫn chính thức: https://www.tensorflow.org/get_started/summaries_and_tensorboard và ví dụ tuyệt vời này của MNIST với tóm tắt: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/tutorials/mnist/mnist_with_summaries.py