print " Done."
if BuildModel:
print "Output Directory:", MyModel.OutDir
# Store the Configuration Dictionary
MyModel.MetaData["Configuration"] = Config
if "HyperParamSet" in dir():
MyModel.MetaData["HyperParamSet"] = HyperParamSet
# Print out the Model Summary
MyModel.Model.summary()
# Compile The Model
print "Compiling Model."
MyModel.BuildOptimizer(optimizer, Config)
MyModel.Compile(Metrics=["accuracy"])
# Train
if Train or (RecoverMode and FailedLoad):
print "Training."
# Setup Callbacks
# These are all optional.
from DLTools.CallBacks import TimeStopping, GracefulExit
from keras.callbacks import *
callbacks = []
# Still testing this...
if TestDefaultParam("UseGracefulExit", 0):
print "Adding GracefulExit Callback."
callbacks.append(GracefulExit())
ReconstructionModel.Build()
print " Done."
print "Output Directory:", ReconstructionModel.OutDir
# Store the Configuration Dictionary
ReconstructionModel.MetaData["Configuration"] = Config
if "HyperParamSet" in dir():
ReconstructionModel.MetaData["HyperParamSet"] = HyperParamSet
# Print out the Model Summary
ReconstructionModel.Model.summary()
# Compile The Model
print "Compiling Model."
ReconstructionModel.BuildOptimizer(optimizer, Config)
ReconstructionModel.Compile(Metrics=["accuracy"])
# Train
if Train or (RecoverMode and FailedLoad):
print "Training."
# Setup Callbacks
# These are all optional.
from DLTools.CallBacks import TimeStopping, GracefulExit
from keras.callbacks import *
callbacks = []
# Still testing this...
if TestDefaultParam("UseGracefulExit", 0):
print "Adding GracefulExit Callback."
callbacks.append(GracefulExit())