Skip to content
Snippets Groups Projects
Commit b43602fa authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 28fa22e0
No related branches found
No related tags found
No related merge requests found
...@@ -519,14 +519,16 @@ class Class1NeuralNetwork(object): ...@@ -519,14 +519,16 @@ class Class1NeuralNetwork(object):
use_multiprocessing=False, use_multiprocessing=False,
workers=1, workers=1,
validation_data=(validation_x_dict, validation_y_dict), validation_data=(validation_x_dict, validation_y_dict),
verbose=verbose,
callbacks=[keras.callbacks.EarlyStopping( callbacks=[keras.callbacks.EarlyStopping(
monitor="val_loss", monitor="val_loss",
patience=patience, patience=patience,
verbose=1)] verbose=verbose)]
) )
if verbose > 0: if verbose > 0:
print("fit_generator completed in %0.2f sec (%d total points)" % ( print("fit_generator completed in %0.2f sec (%d total points)" % (
time.time() - start, yielded_values_box[0])) time.time() - start, yielded_values_box[0]))
return result
def fit( def fit(
......
...@@ -221,7 +221,11 @@ def worker_init_entry_point( ...@@ -221,7 +221,11 @@ def worker_init_entry_point(
def worker_init(keras_backend=None, gpu_device_nums=None, worker_log_dir=None): def worker_init(keras_backend=None, gpu_device_nums=None, worker_log_dir=None):
if worker_log_dir: if worker_log_dir:
sys.stderr = sys.stdout = open( sys.stderr = sys.stdout = open(
os.path.join(worker_log_dir, "LOG-worker.%d.txt" % os.getpid()), "w") os.path.join(worker_log_dir, "LOG-"
""
""
""
"worker.%d.txt" % os.getpid()), "w")
# Each worker needs distinct random numbers # Each worker needs distinct random numbers
numpy.random.seed() numpy.random.seed()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment