tensorboard_server.h

Go to the documentation of this file.

#ifndef TENSORBOARD_SERVER_H
#define TENSORBOARD_SERVER_H
 
#include "bitrl/bitrl_consts.h"
#include "bitrl/bitrl_types.h"
 
#include <boost/noncopyable.hpp>
#include <string>
#include <string_view>
#include <unordered_map>
 
namespace bitrl
{
namespace network
{
 
 
class TensorboardServer : private boost::noncopyable
{
  public:
    TensorboardServer(const std::string &api_url);
 
    void init(const std::string &log_dir);
 
    void close();
 
    void add_scalar(const std::string &tag, real_t value,
                    uint_t step_idx = bitrl::consts::INVALID_ID) const;
 
    void add_scalars(const std::string &main_tag,
                     const std::unordered_map<std::string, real_t> &values,
                     uint_t step_idx = bitrl::consts::INVALID_ID) const;
 
    void add_text(const std::string &tag, const std::string &text,
                  uint_t step_idx = bitrl::consts::INVALID_ID) const;
 
    std::string_view get_log_dir_path() const noexcept { return log_dir_; }
 
    std::string_view get_server_address() const noexcept { return api_url_; }
 
  private:
    std::string api_url_;
    std::string log_dir_;
 
    const std::string api_str_;
};
 
} // namespace utils
} // namespace bitrl
#endif // TENSORBOARD_SERVER_H