PetrochukM / PyTorch-NLP

Basic Utilities for PyTorch Natural Language Processing (NLP)
https://pytorchnlp.readthedocs.io
BSD 3-Clause "New" or "Revised" License
2.21k stars 258 forks source link

Fix `fork_rng_wrap` #108

Open PetrochukM opened 4 years ago

PetrochukM commented 4 years ago

fork_rng_wrap does not propagate function arguments fork_rng_wrap, and it should.