__init__.py 239 字节
Newer Older
W
wangxiao1021 已提交
1 2 3 4 5 6 7 8 9
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()))

from reader import yield_pieces, data_feeder