wandb

1
import pathlib
2

3
import numpy as np
4
import torch
5
import wandb
6

7
run = wandb.init(name=pathlib.Path(__file__).stem)
8
run.log({"cuda_available": torch.cuda.is_available()})
9
x = np.random.random((32, 100)).astype("f")
10
t_cpu = torch.Tensor(x)
11
t_gpu = t_cpu.cuda()
12

13
run.log({"host_tensor": t_cpu})
14
run.log({"cuda_tensor": t_gpu})
15

16
run.finish()
17