![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch学习
cintinueFAn
这个作者很懒,什么都没留下…
展开
-
FCOS源码阅读1
FCOS代码真的好多啊,一点一点分析好了,先来看看tools目录下的train_net.pydef train(cfg, local_rank, distributed): #local rank是啥,distributed就是给那些gpu暴发户用的吧 #得到一个model,这个函数在下一篇文章中看 model = build_detection_model(cfg) #torch.device代表将torch.Tensor分配到的设备的对象。 device = torch.d原创 2020-05-24 17:40:34 · 780 阅读 · 0 评论 -
torch.squeeze()和unsqueeze()
queeze()函数功能:去除size为1的维度,包括行和列。当维度大于等于2时,squeeze()无作用。其中squeeze(0)代表若第一维度值为1则去除第一维度,squeeze(1)代表若第二维度值为1则去除第二维度。eg1:a = torch.Tensor(1,3)print a tensor([[-1.37,4.56,-3.57]])...转载 2018-08-22 11:43:44 · 27034 阅读 · 0 评论 -
matplotlib中的ion()和ioff()函数
在学习莫烦python的一些代码时发现经常出现plt.ion()和plt.ioff函数,经查询总结如下:python数据可视化库matplotlib有两种运行模式阻塞(block)模式和交互(interactive)模式,在python console中默认为交互模式,而在python脚本中是阻塞模式。1.在阻塞模式下,plt.plot()和plt.imshow()需要plt.show()...原创 2018-08-23 15:50:22 · 2784 阅读 · 0 评论