Skip to content

Commit

Permalink
remove redundant annotations
Browse files Browse the repository at this point in the history
  • Loading branch information
HydrogenSulfate committed Dec 21, 2023
1 parent 4991b3d commit eecad82
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions deepmd/utils/batch_size.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,11 +100,9 @@ def execute(
OOM when batch size is 1
"""
try:
# print(__file__, self.current_batch_size, natoms)
n_batch, result = callable(
max(self.current_batch_size // natoms, 1), start_index
)
# print(__file__, n_batch)
except OutOfMemoryError as e:
# TODO: it's very slow to catch OOM error; I don't know what TF is doing here
# but luckily we only need to catch once
Expand Down

0 comments on commit eecad82

Please sign in to comment.