Commit Graph

19 Commits (develop)

Author SHA1 Message Date
Kaipeng Deng a32e8bf1e7
DataLoader supprot dict str (#31481)
5 years ago
Kaipeng Deng c4ddc3ab0d
fix dataloader collate return list mix tensor and numpy array (#30904)
5 years ago
Kaipeng Deng 302427170f
remove numpy array check in single-process dataloader. test=develop (#30861)
5 years ago
Leo Chen 3d015f1cf5
Set expected place in child thread for dataloader to avoid costing cuda memory on other card (#30338)
5 years ago
mls1999725 0aedd463ee
Update get_worker_info API (#29190)
5 years ago
Leo Chen 3815d7aa40
Upgrade string literals to raw string (#28989)
5 years ago
Kaipeng Deng 89d27de90f
DataLoader support not auto collate batch (#28425)
5 years ago
Kaipeng Deng 4671d85a03
fix DataLoader return same format between static & dynamic in single mode (#28176)
6 years ago
LielinJiang 8327accc58
Fix dataloader when stack input data with different type (#27950)
6 years ago
Kaipeng Deng d17681dcc7
fix DataLoader single process mode exit SIGABRT error. (#27850)
6 years ago
Chen Weihang 8d53172794
move DataLoader._worker_loop to top level (#27247)
6 years ago
Kaipeng Deng c03092b188
fix dataloader performace decrease & unittest hanging. test=develop (#26739)
6 years ago
Kaipeng Deng 84e7be3178
add TensorDataset for multiprocess DataLoader (#26332)
6 years ago
Kaipeng Deng 7051bbc2a1
fix test_multiprocess_dataloader unittest. test=develop (#26241)
6 years ago
Kaipeng Deng dbc88bb900
Add iterable dataset support for multiprocess DataLoader (#25558)
6 years ago
Chen Weihang 3c8daa9b89
Add pin memory control for BufferedReader (#26026)
6 years ago
Kaipeng Deng 20ee36bdea
make default_collate_fn visible. test=develop (#25244)
6 years ago
Kaipeng Deng 826c607e0c
Fix test multiprocess dataloader static (#25287)
6 years ago
Kaipeng Deng 80cf3c3c4d
Refine DataLoader support multi-processing (#23107)
6 years ago