From 3f2c6d18a36654e8e9198e6e8deb0be34c1adab6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Matt=C3=A9o=20Delabre?= Date: Mon, 16 Dec 2019 01:50:03 -0500 Subject: [PATCH] Manually seed training for reproducibility --- train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train.py b/train.py index f6d274e..87fad5e 100644 --- a/train.py +++ b/train.py @@ -4,6 +4,8 @@ import torch from matplotlib import pyplot from mpl_toolkits.axisartist.axislines import SubplotZero +torch.manual_seed(42) + # Number of symbols to learn order = 4