实操过程当中的pytorch小知识点

pytorch模型训练特征图可视化 参考： １ ２html tensorflow使用tensorboard来可视化，pytorch使用tensorboardX来可视化。须要安装。 使用cuda训练 当单个cuda x = torch.Tensor([1,2,3]).cuda 当多个cuda,不指定时默认使用cuda0 x = torch.Tensor([1,2,3],device = torch.

>>阅读原文<<