Tensorflow provides golang api for model training and inference, however currently tensorboard is only supported when using python.
This repository enables using tensorboard with tensorflow golang api. tfsum.Writer writes file that tensorboard could understand.
Before using the following piece of code, you have to build tensorflow golang api: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/go/README.md
package main
import (
"fmt"
"github.com/helinwang/tfsum"
tf "github.com/tensorflow/tensorflow/tensorflow/go"
)
func main() {
step := 0
w := &tfsum.Writer{Dir: "./tf-log", Name: "train"}
var s *tf.Session
var g *tf.Graph
var input *tf.Tensor
sum, err := s.Run(
map[tf.Output]*tf.Tensor{
g.Operation("input").Output(0): input,
},
[]tf.Output{
g.Operation("MergeSummary/MergeSummary").Output(0),
},
[]*tf.Operation{
g.Operation("train_step"),
})
if err != nil {
fmt.Println(err)
}
err = w.AddEvent(sum[0].Value().(string), int64(step))
if err != nil {
fmt.Println(err)
}
}
Then run tensorboard normally
tensorboard --logdir=./tf-log
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent