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 merge requests found
......@@ -519,14 +519,16 @@ class Class1NeuralNetwork(object):
use_multiprocessing=False,
workers=1,
validation_data=(validation_x_dict, validation_y_dict),
verbose=verbose,
callbacks=[keras.callbacks.EarlyStopping(
monitor="val_loss",
patience=patience,
verbose=1)]
verbose=verbose)]
)
if verbose > 0:
print("fit_generator completed in %0.2f sec (%d total points)" % (
time.time() - start, yielded_values_box[0]))
return result
def fit(
......
......@@ -221,7 +221,11 @@ def worker_init_entry_point(
def worker_init(keras_backend=None, gpu_device_nums=None, worker_log_dir=None):
if worker_log_dir:
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
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