• J
    dataset (#17973) · 3f8031e2
    jiaqi 提交于
    (1) use channel instead of vector/BlockingQueue in Dataset,to keep same with existing implementation, and make code more readable and flexible (dataset single output channel or multi output channel). one previous memory out of limit problem is cause by not release memory after training.
    (2) add Record because MultiSlotType costs too much memory (80B),fix memory out of limit problem.
    (3) add Channel, Archive in paddle/fluid/framework
    (4) change dataset from shared_ptr to unique_ptr in pybind
    (5) move create/destroy readers from trainer to dataset
    (6) move shuffle from datafeed to dataset. dataset holds memory, datafeed is only for load data and feed data to network.
    (7) fix thread num bug of Dataset when filelist size < thread num
    (8) support set_queue_num in InMemoryDataset
    3f8031e2
multi_trainer.cc 2.6 KB