如何解决如何确保Pytorch模型的结果跨设备的可重复性?
我对pytorch模型的随机性有些困惑。
我使用以下代码修复了随机种子,以便可以在同一设备上重复模型的训练结果:
def seed_torch(seed=2020):
random.seed(seed)
os.environ["PYTHONHASHSEED"] = str(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
但是令人困惑的是,当代码在具有相同硬件配置和软件环境的另一台设备上运行时,会产生不同的结果。
我重新安装了虚拟环境,以确保库的版本一致,但是这个问题仍然使我感到困惑。
随时询问是否需要更多代码来解释问题。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。