Open Edouard127 opened 2 years ago
You can either chose to save it in .h5 or .pb
model.save("nsfw." + str(width) + "x" + str(height)) for .pb It creates a folder with assets keras_metadata.pb saved_model.pb variables
model.save("nsfw." + str(width) + "x" + str(height))
assets keras_metadata.pb saved_model.pb variables
And model.save("nsfw." + str(width) + "x" + str(height)) + ".h5" for .h5
model.save("nsfw." + str(width) + "x" + str(height)) + ".h5"
You can either chose to save it in .h5 or .pb
model.save("nsfw." + str(width) + "x" + str(height))for .pb It creates a folder withassets keras_metadata.pb saved_model.pb variablesAnd
model.save("nsfw." + str(width) + "x" + str(height)) + ".h5"for .h5