Skip to content

Commit

Permalink
Merge branch 'distconv-python-data-reader' of github.com:fiedorowicz1…
Browse files Browse the repository at this point in the history
…/lbann into distconv-python-data-reader
  • Loading branch information
fiedorowicz1 committed Mar 25, 2024
2 parents 3310e64 + 4d94f0c commit e0cf5f2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/lbann/util/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -271,7 +271,7 @@ def get_batch(self, batch_size: int) -> Dict[str, Union[np.ndarray, int]]:

batch = {}

# Note: we return the arrays wiht the pointers so that they aren't
# Note: we return the arrays with the pointers so that they aren't
# deallocated by the garbage collector.
batch["sample"] = np.ascontiguousarray(
[s.sample for s in samples], dtype=self.dtype
Expand Down

0 comments on commit e0cf5f2

Please sign in to comment.