Manually seed training for reproducibility

2019-12-16 01:50:03 -05:00 · 2019-12-16 01:50:03 -05:00 · 3f2c6d18a3
parent 4d8d419d31
commit 3f2c6d18a3
1 changed files with 2 additions and 0 deletions
--- 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