python 程序中如何保存随机数状态?

在原来的程序里增加了一段代码，这段代码影响了随机数生成（原因不明），导致以前的结果无法复现。有什么办法能在这段代码之前保存随机数的状态，之后再恢复原来的状态？

补充：增加的代码是调用 pytorch 跑一个卷积网络。程序用到了 numpy 和 random。