matteo/constellationnet
matteo
/
constellationnet
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Manually seed training for reproducibility

This commit is contained in:
Mattéo Delabre 2019-12-16 01:50:03 -05:00
parent 4d8d419d31
commit 3f2c6d18a3
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -4,6 +4,8 @@ import torch
from matplotlib import pyplot from matplotlib import pyplot
from mpl_toolkits.axisartist.axislines import SubplotZero from mpl_toolkits.axisartist.axislines import SubplotZero
torch.manual_seed(42)
# Number of symbols to learn # Number of symbols to learn
order = 4 order = 4