自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(6)
  • 收藏
  • 关注

原创 Logging中的format

最近在代码中看到logging中的formatFORMAT = "[%(asctime)-15s %(filename)s:%(lineno)d %(funcName)s] %(message)s"logging.basicConfig(format=FORMAT)logger_vis = logging.getLogger(__name__)logger_vis.setLevel(logging.DEBUG)由于之前看的代码比较简单,没关注过这个东西,查看资料后发现这是用于显示日志的。Log

2021-06-02 20:03:29 1119

原创 cuda runtime error (38) : no CUDA-capable device is detected

在vscode中训练模型时,出现如下错误:cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:51原因在于CUDA没选对,需要在先查看自己有几块显卡:在terminal中输入,nvcc --version # 查看自己的CUDA版本...

2021-06-01 15:58:59 2286

原创 unsupported operand type(s) for +: ‘NoneType‘ and ‘str‘

在进行创建输出结果的路径时发生错误:unsupported operand type(s) for +: ‘NoneType’ and ‘str’代码展示如下:tn = args.ttask_name = args.data_dir.split('/')[-2] + '/' + args.data_dir.split('/')[-1]train_result_path = osp.join('result',task_name,'train',args.name + '_' +str(args.l

2021-06-01 15:04:49 2132

原创 args = parser.parse_args() 出现问题

直接调试出现问题描述error: the following arguments are required: -d/–data-dir, -j/–workersDebug出现问题描述Exception has occurred: SystemExit 2解决:1、可以在terminal中直接运行,-d/–data-dir=‘your path’, -j/–workers=‘your name’2、若要debug,default定义好,把required去掉...

2021-06-01 14:38:20 2001 3

原创 基于PyTorch框架的torchvision

获取数据集在图像分类最常用的手写数字识别数据集MNIST上作为例子进行学习基于PyTorch框架的torchvision:torchvision.datasets:下载数据集,包含一些加载数据的函数及数据集接口包含的数据集有:MNIST、COCO(用于图像标记和目标检测)、LSUN ClassificationImageFolder、Imagenet-12、CIFAR10 and CIFAR100、STL10其中Dataset拥有__getitem__和__len__两个API(接口),可以

2021-04-22 14:46:16 131 2

原创 name ‘device‘ is not defined

在训练简单的CCN进行模型选择时,出现name ‘device’ is not defined发现是没有定义device,因此查看了以下model = model.to(device)这行代码的意义,https://blog.csdn.net/weixin_36670529/article/details/104367696博客中描述的很详细**,是说将模型加载到指定设备上**。其中,device=torch.device(“cpu”)代表的使用cpu,而device=torch.device(“cu

2021-04-20 15:51:44 7894 1

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除