pytorch
hyluglare
这个作者很懒,什么都没留下…
展开
-
detectron学习记录
刚开始学习,不断更新中… 【安装】 github地址 注意避坑: torchvision和pytorch一起在官网安装好 python 3.6以上 cuda驱动提前安装好 通过本地安装: git clone https://github.com/facebookresearch/detectron2.git cd detectron2 pip3 install -e . 下载预训练模型:官网...原创 2020-02-23 12:16:02 · 241 阅读 · 0 评论 -
pytorch中tensor的常用操作
近日开始学习pytorch编程,对其中常用到的tensor操作进行简单整理,方便自己查取,如果更好的方法,欢迎留言。 tensor的初始化 首先定义tensor的大小,记为size # 支持多维的size size = (dim1, dim2, ...) 常用的初始化方法 # a 返回大小为size的一个空tensor a = torch.empty(size) # b 返回大小为size...原创 2019-03-14 13:16:10 · 2229 阅读 · 0 评论