From dd5ec3dbde92410e26a5c7907d8657e5daff358b Mon Sep 17 00:00:00 2001
From: Connor Olding <cloningdonor@gmail.com>
Date: Sat, 9 Jun 2018 16:20:07 +0200
Subject: [PATCH] make network linear

---
 main.lua | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.lua b/main.lua
index 004fefd..82e1ba0 100644
--- a/main.lua
+++ b/main.lua
@@ -150,6 +150,7 @@ local function make_network(input_size)
     nn_x:feed(nn_y)
     nn_ty:feed(nn_y)
 
+    --[[
     nn_y = nn_y:feed(nn.Dense(128))
     if cfg.deterministic then
         nn_y = nn_y:feed(nn.Relu())
@@ -157,6 +158,7 @@ local function make_network(input_size)
         nn_y = nn_y:feed(nn.Gelu())
     end
     if cfg.layernorm then nn_y = nn_y:feed(nn.LayerNorm()) end
+    --]]
 
     nn_z = nn_y
     nn_z = nn_z:feed(nn.Dense(#gcfg.jp_lut))