5 Commits (10a5f38aaa0d4e86b542dfad954d15eab8d5c6de)

Author	SHA1	Message	Date
xyliao	10a5f38aaa	fix(dataloader): delete background threading when dataloader is freed (#583)	2021-10-01 16:14:36 +08:00
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