https://github.com/palle-k/dl4s-tensorboard
Pure Swift TensorBoard plugin for DL4S
https://github.com/palle-k/dl4s-tensorboard
deep-learning deep-learning-visualization swift tensorboard
Last synced: 3 months ago
JSON representation
Pure Swift TensorBoard plugin for DL4S
- Host: GitHub
- URL: https://github.com/palle-k/dl4s-tensorboard
- Owner: palle-k
- License: mit
- Created: 2020-06-02T02:32:21.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2021-11-09T01:20:37.000Z (over 4 years ago)
- Last Synced: 2026-02-15T01:28:22.592Z (4 months ago)
- Topics: deep-learning, deep-learning-visualization, swift, tensorboard
- Language: Swift
- Homepage:
- Size: 43 KB
- Stars: 2
- Watchers: 1
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# DL4S-Tensorboard
A pure Swift summary writer for [DL4S](https://github.com/palle-k/DL4S) that writes Tensorboard log files.
### Installation
Add DL4S-Tensorboard as a package dependency in `Package.swift`.
.package(url: "https://github.com/palle-k/DL4S.git", .branch("master")),
.package(url: "https://github.com/palle-k/DL4S-Tensorboard.git", .branch("master"))
Then add `DL4STensorboard` as a target dependency:
.target(name: "YourAwesomeTarget", dependencies: ["DL4S", "DL4STensorboard"]),
### Usage
DL4S-Tensorboard supports Scalars, Images, Text, Embeddings and Histograms.
#### Writing scalars
```swift
import DL4STensorboard
import Foundation
let logdir = URL(fileURLWithPath: "./logs")
let writer = try TensorboardWriter(logDirectory: logdir, runName: "Classifier")
try writer.write(scalar: 101, withTag: "model/accuracy", atStep: 1337)
```
#### Advanced Usage
```swift
// writing an image
try writer.write(image: imageTensor, withTag: "generator/output", atStep: 42)
// writing text
try writer.write(text: "Lorem ipsum dolor sit amet", withTag: "lm/sample", atStep: 314)
// writing embeddings
let embeddingLayer = DL4S.Embedding(inputFeatures: 42, outputSize: 128)
try writer.write(embedding: embeddingLayer.embeddingMatrix, withLabels: vocab, atStep: 1337)
// writing a histogram
let histogram = Histogram(values: valueDistribution, buckets: 10)
try writer.write(histogram: histogram, withTag: "data/histogram", atStep: 4242)
```