3.2 KiB
3.2 KiB
In [ ]:
!pip install tensorboard
In [12]:
%load_ext tensorboard %tensorboard --logdir runs --host 0.0.0.0 --port 6006
The tensorboard extension is already loaded. To reload it, use: %reload_ext tensorboard
Reusing TensorBoard on port 6006 (pid 414), started 0:24:42 ago. (Use '!kill 414' to kill it.)
<script>
(function() {
const frame = document.getElementById("tensorboard-frame-42c02cf9641fd4a7");
const url = new URL("/", window.location);
const port = 6006;
if (port) {
url.port = port;
}
frame.src = url;
})();
</script>
In [11]:
from torch.utils.tensorboard import SummaryWriter import numpy as np # 创建一个SummaryWriter对象 writer = SummaryWriter() for n_iter in range(100): writer.add_scalar('Loss/train', np.random.random(), n_iter) writer.add_scalar('Loss/test', np.random.random(), n_iter) writer.add_scalar('Accuracy/train', np.random.random(), n_iter) writer.add_scalar('Accuracy/test', np.random.random(), n_iter) img = np.zeros((3, 100, 100)) img[0] = np.arange(0, 10000).reshape(100, 100) / 10000 img[1] = np.arange(0, 10000).reshape(100, 100) / 10000 writer.add_image('my_image', img, 0) writer.close()