o i e@sdZddlZddlmZddlmZddlmZddlmZ zddl Z Wn e y/dZ Ynwej Z e ddddd Zed krQedge rSeedSdSdS) zMemory profile on Keras model. To add a new model for memory profile: 1. Create the model. 2. Decorate it with `@memory_profiler.profile`. 3. Add the model function to the dict `models`. N)app)flags)loggingmodelz!The model to run memory profiler.cCsPtjdd}d|i}tj|vr!tdtj|tj}|dStddS)NcSstjjdddd}tjd}t}|tjdd|tjjdddd |tjj d d d | d d|j ||ddddS)z LSTM model.ri) d)size)ri Ng?)dropoutrecurrent_dropoutr sigmoid) activationsgdmsei) batch_sizeepochsN) nprandomrandintkeras Sequentialaddlayers EmbeddingLSTMDensecompilefit)x_trainy_trainrr"X/var/www/myenv/lib/python3.10/site-packages/keras/src/benchmarks/model_memory_profile.py_imdb_lstm_model)s  zmain.._imdb_lstm_modellstmzRun memory profile on %s.z7The model does not exist. Please verify the model name.)memory_profilerprofileFLAGSrrinfo)_r$models run_modelr"r"r#main(s    r-__main__)__doc__numpyrabslrrr keras.srcsrcrr& ImportErrorr( DEFINE_stringr-__name__mark_flags_as_requiredrunr"r"r"r#s(