hardmaru / WorldModelsExperiments

World Models Experiments
608 stars 171 forks source link

Fixes a leak where new tensorflow assign ops are created when setting model parameters #14

Closed davidsandberg closed 5 years ago

davidsandberg commented 5 years ago

This is a fix for a memory leak where new Tensorflow assign operations are created every time set_model_params(...) is called. The fix consists of creating the assign operations together with placeholders to be able to feed new parameter values to the model.

hardmaru commented 5 years ago

Thanks for the PR!