Update main.py

main.py

@@ -683,8 +683,8 @@ class MyCustomCallback(tf.keras.callbacks.Callback):
 
 k = model.fit(X_train,
           y_train,
-          epochs=10,
-          batch_size=2048,
+          epochs=8,
+          batch_size=4000,
           validation_data=(X_test, y_test),
           callbacks=[MyCustomCallback(),tf.keras.callbacks.EarlyStopping(monitor='val_loss', mode='min', verbose=1, patience=10),tensorboard_callback],
           verbose=1
@@ -809,7 +809,9 @@ import time
 print("MODELLLLLLL save ")
 os.system('cp log.txt old.txt')
 
-time.sleep(99999)
+while 1:
+    print("waiting")
+