Skip to content

PyTorch

在学术研究者当中,PyTorch 是最流行的 Python 深度学习框架。

PyTorch

你可以使用PyTorch进行深度学习模型训练,同时使用SwanLab进行实验跟踪与可视化。

Pytorch生态的其他集成

记录Tensor图像

你可以将带有图像数据的PyTorch Tensors传递给swanlab.Imageswanlab.Image将使用torchvision把它们转换成图像:

python
image_tensors = ...  # shape为[B, C, H, W]的Tensor图像
swanlab.log({"examples": [swanlab.Image(im) for im in image_tensors]})