mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-15 07:09:55 +02:00
do_prediction: trigger GC to avoid CUDA OOM
This commit is contained in:
parent
ad748d0039
commit
d68017037c
1 changed files with 2 additions and 1 deletions
|
@ -999,7 +999,7 @@ class Eynollah:
|
|||
|
||||
prediction_true = prediction_true.astype(np.uint8)
|
||||
#del model
|
||||
#gc.collect()
|
||||
gc.collect()
|
||||
return prediction_true
|
||||
|
||||
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))
|
||||
|
||||
prediction_true = prediction_true.astype(np.uint8)
|
||||
gc.collect()
|
||||
return prediction_true
|
||||
|
||||
def extract_page(self):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue