Skip to content

Commit

Permalink
Update deeplc.py
Browse files Browse the repository at this point in the history
  • Loading branch information
RobbinBouwmeester committed Aug 8, 2024
1 parent 8dff0c1 commit 20a3b7a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions deeplc/deeplc.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,10 +134,10 @@ def divide_chunks(l, n):

def reset_keras():
"""Reset Keras session."""
sess = get_session()
clear_session()
sess.close()
gc.collect()
# sess = get_session()
# clear_session()
# sess.close()
# gc.collect()
# Set to force CPU calculations
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

Expand Down

0 comments on commit 20a3b7a

Please sign in to comment.