4 Commits (1dce15efad4e00c229a722a66894badc76f00da6)

Author	SHA1	Message	Date
Xingyu Liao	1dce15efad	faster dataloader with pre-fetch and cuda stream (#456) ... Summary: add a background thread to create a generator with pre-fetch, and create a new cuda stream to copy tensor from cpu to gpu in parallel. Reviewed by: l1aoxingyu	2021-04-12 15:03:35 +08:00
liaoxingyu	b5c3c0a24d	update docs	2021-01-22 21:11:19 +08:00
liaoxingyu	20a01f2545	fix grayscale image input problem ... Summary: handle grayscale images mixed in RGB images by adding the image channels	2020-12-28 14:45:09 +08:00
L1aoXingyu	db6ed12b14	Update sampler code	2020-02-10 07:38:56 +08:00