首页 PaddleCV 帖子详情
计算机高级api resnet50 的数据格式是什么呀? 已解决
收藏
快速回复
PaddleCV 问答图像分类 564 10
计算机高级api resnet50 的数据格式是什么呀? 已解决
收藏
快速回复
PaddleCV 问答图像分类 564 10

我的数据格式Tensor[(shape=[3, 224, 224], dtype=float64, place=CPUPlace, stop_gradient=True),label],继承了iodataset的类,

但是报错为

(InvalidArgument) input and filter data type should be consistent, but received input data type is double and filter type is float
[Hint: Expected input_data_type == filter_data_type, but received input_data_type:6 != filter_data_type:5.] (at /paddle/paddle/fluid/operators/conv_op.cc:211)
[operator < conv2d > error]

老师们,到底怎么解决这个问题

FutureSI
已解决
2# 回复于2022-03
一般应该用float32类型
0
收藏
回复
全部评论(10)
时间顺序
FutureSI
#2 回复于2022-03

一般应该用float32类型

1
回复
FutureSI
#3 回复于2022-03

输入的tensor是 NCHW 格式

0
回复
Nefelibata0
#4 回复于2022-03

谢谢老师

0
回复
Nefelibata0
#6 回复于2022-03
输入的tensor是 NCHW 格式

请问老师,在使用高级API时,如何使用GPU来训练?

0
回复
Nefelibata0
#7 回复于2022-03
输入的tensor是 NCHW 格式

已经解决了,谢谢老师

0
回复
fi_Past
#8 回复于2022-03

gpu会自动有用吧

0
回复
fi_Past
#9 回复于2022-03

和torch不太一样

0
回复
fi_Past
#10 回复于2022-03

开始有点不适应

0
回复
Nefelibata0
#11 回复于2022-03
fi_Past #8
gpu会自动有用吧

好像需要下面一段代码

import paddle.fluid as fluid

use_gpu = True place = fluid.CUDAPlace(0) if use_gpu else fluid.CPUPlace()

0
回复
滑板鞋
#12 回复于2022-04

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户