PMPBinZhang comments

Repositories
Issues
Comments

Results 1 comments of


PMPBinZhang

您好，paddlex怎么指定设备训练呢，我安装的是paddlepaddle是gpu版本，但是我想用CPU进行训练，采用在import paddlex前使用os.environ['CUDA_VISIBLE_DEVICES'] = ''和使用paddle.set_device("cpu")都没有起作用。

您好，我训练的时候train_batch_size给2，就报错(NotFound) Operator softmax does not have kernel for {data_type[int64_t]; data_layout[Undefined(AnyLayout)]; place[Place(cpu)]; library_type[PLAIN]}. [Hint: Expected kernel_iter != kernels.end(), but received kernel_iter == kernels.end().] (at ..\paddle\fluid\imperative\prepared_operator.cc:419) [operator < softmax > error]，给4就没有这种情况，这是什么原因呢 |...