qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-11-04 03:34:16 +01:00
Code Issues Projects Releases Wiki Activity

OOM error happend with tensorflow-gpu=1.15.5 is resolved

Browse source
This commit is contained in:
vahid 2021-04-01 11:22:08 -04:00
parent 67a9fc8820
commit b473c85a59
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
qurator/eynollah/eynollah.py
View file

@ -429,7 +429,7 @@ class Eynollah:
self.writer.height_org = self.height_org self.writer.height_org = self.height_org
self.writer.width_org = self.width_org self.writer.width_org = self.width_org
def start_new_session_and_model(self, model_dir): def start_new_session_and_model_old(self, model_dir):
self.logger.debug("enter start_new_session_and_model (model_dir=%s)", model_dir) self.logger.debug("enter start_new_session_and_model (model_dir=%s)", model_dir)
config = tf.ConfigProto() config = tf.ConfigProto()
config.gpu_options.allow_growth = True config.gpu_options.allow_growth = True
@ -438,7 +438,13 @@ class Eynollah:
model = load_model(model_dir, compile=False) model = load_model(model_dir, compile=False)
return model, session return model, session
def start_new_session_and_model(self, model_dir):
self.logger.debug("enter start_new_session_and_model (model_dir=%s)", model_dir)
gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=7.7, allow_growth=True)
session = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
model = load_model(model_dir, compile=False)
return model, session
def do_prediction(self, patches, img, model, marginal_of_patch_percent=0.1): def do_prediction(self, patches, img, model, marginal_of_patch_percent=0.1):
self.logger.debug("enter do_prediction") self.logger.debug("enter do_prediction")