更改工作目录

您正在使用一个开源库来在您的数据上训练深度神经网络。不巧的是，训练过程中，该库会将检查点模型（即仅在部分数据上训练过的模型）写入当前工作目录。这个行为让您很困扰，因为您并不想总是从将要保存模型的目录来启动脚本。

您决定编写一个上下文管理器来解决这个问题：进入上下文时更改当前工作目录，允许您构建模型，退出时再将工作目录重置到原来的位置。您需要确保在模型训练过程中即使发生错误，也不会影响将工作目录重置回初始位置。