https://github.com/helinwang/tfsum
Enable TensorBoard for TensorFlow Go API
https://github.com/helinwang/tfsum
golang tensorboard tensorboard-visualizations tensorflow
Last synced: about 1 year ago
JSON representation
Enable TensorBoard for TensorFlow Go API
- Host: GitHub
- URL: https://github.com/helinwang/tfsum
- Owner: helinwang
- License: bsd-3-clause
- Created: 2016-12-11T05:12:10.000Z (over 9 years ago)
- Default Branch: master
- Last Pushed: 2017-04-24T20:50:08.000Z (about 9 years ago)
- Last Synced: 2025-03-24T05:51:58.988Z (about 1 year ago)
- Topics: golang, tensorboard, tensorboard-visualizations, tensorflow
- Language: Go
- Homepage:
- Size: 56.6 KB
- Stars: 33
- Watchers: 1
- Forks: 6
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
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
### Example
```
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
```