计算机高级api resnet50 的数据格式是什么呀?
收藏
我的数据格式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]
老师们,到底怎么解决这个问题
0
收藏
请登录后评论
一般应该用float32类型
输入的tensor是 NCHW 格式
谢谢老师
请问老师,在使用高级API时,如何使用GPU来训练?
已经解决了,谢谢老师
gpu会自动有用吧
和torch不太一样
开始有点不适应
好像需要下面一段代码
import paddle.fluid as fluid
use_gpu = True place = fluid.CUDAPlace(0) if use_gpu else fluid.CPUPlace()
赞