From 9a65441095caef62bad8c0b2a209983b08dbd983 Mon Sep 17 00:00:00 2001
From: Sergii Dymchenko <kit1980@gmail.com>
Date: Tue, 2 Jun 2020 03:20:19 -0700
Subject: [PATCH] Don't use 1 epoch for tuning.

---
 luxor/tune.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/luxor/tune.py b/luxor/tune.py
index 903fa50..9c7157e 100644
--- a/luxor/tune.py
+++ b/luxor/tune.py
@@ -22,7 +22,6 @@ else:
 
 def train_wrapper(parameterization):
     net = Net()
-    parameterization["num_epochs"] = 1
     last_epoch_loss = train(net=net, data_loader=train_loader, parameters=parameterization, device=device)
     if math.isnan(last_epoch_loss):
         last_epoch_loss = 10**6