Pytorch基本操作

Pytorch的通道顺序是NCHW

Tensor

是pytorch使用的基本数据结构，可以在GPU上、也可以在CPU上运行。（numpy的ndarray只能在cpu上运行）

创建或将数据转化为tensor:

torch.tensor(data,dtype,device) #创建tensor， data可以是python list或者np.array， dtype可以是torch.float32, torch.float64,torch.int8等）
torch.as_tensor(data, dtype,device)  #把data转化为tensor，如果data已经是tensor且dtype和device与data一致，则不发生copy；如果data是ndarray且dtype和device与data一致，则不发生copy

TensorboardX

用于pytorch训练过程可视化loss曲线。

tensorboardX依赖于tensorboard。提前安装：

pip3 install tensorboard==1.14.0
pip3 install tensorboardX

备注：tensorboard版本升级：

pip3 install tensorboard --upgrade

 

使用：

tensorboard --logdir=./path_to_log

弹出：

TensorBoard 1.14.0 at http://yly:6006/

网页打开该地址即可看到结果。