纯净、安全、绿色的下载网站

首页|软件分类|下载排行|最新软件|IT学院

当前位置:首页IT学院IT技术

pytorch-yolov3 train报错 解决pytorch-yolov3 train 报错的问题

bbzz2   2021-04-21 我要评论

UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead

https://github.com/eriklindernoren/PyTorch-YOLOv3/blob/master/models.py

将model.py 

obj_mask转为int8 bool

obj_mask=obj_mask.bool() # convert int8 to bool

noobj_mask=noobj_mask.bool() #convert int8 to bool 

以上这篇解决pytorch-yolov3 train 报错的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。


相关文章

猜您喜欢

  • python chardet Python chardet库识别编码原理解析

    想了解Python chardet库识别编码原理解析的相关内容吗,在本文为您仔细讲解python chardet的相关知识和一些Code实例,欢迎阅读和指正,我们先划重点:python,chardet,库,下面大家一起来学习吧。..
  • pyecharts安装失败 Python中pyecharts安装及安装失败的解决办法

    想了解Python中pyecharts安装及安装失败的解决办法的相关内容吗,lb990123在本文为您仔细讲解pyecharts安装失败的相关知识和一些Code实例,欢迎阅读和指正,我们先划重点:pyecharts安装失败,pyecharts安装,下面大家一起来学习吧。..

网友评论

Copyright 2020 www.marker-soft.com 【马克软件站】 版权所有 软件发布

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 点此查看联系方式