[Enhance] Allow users to customize worker_init_fn of Dataloader (#1038)
* customize worker init fn function * add assert * narrow worker_init_fn type
Loading
Please register or sign in to comment
* customize worker init fn function * add assert * narrow worker_init_fn type