PyTorch-NLP
PyTorch-NLP copied to clipboard
Fix `fork_rng_wrap`
fork_rng_wrap
does not propagate function arguments fork_rng_wrap
, and it should.