__init__.py 251 字节
Newer Older
W
wangxiao1021 已提交
1 2 3 4 5 6 7
from paddle import fluid
import os
import multiprocessing

gpu_dev_count = int(fluid.core.get_cuda_device_count())
cpu_dev_count = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count()))

W
wangxiao1021 已提交
8
from reader import yield_pieces, data_feeder, decode_fake
W
wangxiao1021 已提交
9