double epochs

justaviet · Jan 19, 2017 · ec96bf0 · ec96bf0
1 parent 5ffc96f
commit ec96bf0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/speednet.py b/speednet.py
@@ -2,6 +2,7 @@
 from keras.layers import Dropout, Flatten, Dense, Activation
 from keras.layers.convolutional import Convolution2D, MaxPooling2D
 from keras.optimizers import Adam
+from keras.callbacks import EarlyStopping
 import numpy as np
 import cv2
 import sys
@@ -15,7 +16,7 @@ class SpeedNet:
 
     DSIZE = (100,100)
     W_FILE = "weights.h5"
-    EPOCHS = 50
+    EPOCHS = 100
     BATCH_SIZE = 32
 
     def main(self, args):
@@ -164,6 +165,7 @@ def train(self,X_src,Y_src,val_split, wipe,n_epochs = 50, batch_size= 32):
         #load data
         X,Y = self.prep_data(X_src,Y_src,shuffle = True,wipe = wipe)
         X = X[:,:,:,[0,2]] #extract channels with data
+
         #train model
         print "Starting training"
         self.model.fit(X, Y, batch_size=batch_size,

diff --git a/weights.h5 b/weights.h5