mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-08 19:59:56 +02:00
OOM error happend with tensorflow-gpu=1.15.5 is resolved
This commit is contained in:
parent
67a9fc8820
commit
b473c85a59
1 changed files with 7 additions and 1 deletions
|
@ -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")
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue