qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-15 23:29:54 +02:00
Code Issues Projects Releases Wiki Activity

do_prediction: trigger GC to avoid CUDA OOM

Browse source
This commit is contained in:
Robert Sachunsky 2024-12-09 11:27:11 +00:00
parent ad748d0039
commit d68017037c
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/eynollah.py
View file

@ -999,7 +999,7 @@ class Eynollah:
prediction_true = prediction_true.astype(np.uint8) prediction_true = prediction_true.astype(np.uint8)
#del model #del model
#gc.collect() gc.collect()
return prediction_true return prediction_true
def do_padding_with_scale(self,img, scale): def do_padding_with_scale(self,img, scale):
@ -1209,6 +1209,7 @@ class Eynollah:
img_patch = np.zeros((n_batch_inference, img_height_model, img_width_model, 3)) img_patch = np.zeros((n_batch_inference, img_height_model, img_width_model, 3))
prediction_true = prediction_true.astype(np.uint8) prediction_true = prediction_true.astype(np.uint8)
gc.collect()
return prediction_true return prediction_true
def extract_page(self): def extract_page(self):