tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-06 04:12:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

further cleanup

Browse Source
This commit is contained in:
Sofie Van Landeghem 2025-11-22 22:50:34 +01:00 committed by GitHub
parent df9a644e24
commit c09b897601
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tasks/common.py
View File

@ -37,16 +37,15 @@ class Task:
def __len__(self):
start = self.start
if self.stop is not None:
num_ex = self.num_examples()
if self.stop > num_ex:
num_ex = self.num_examples()
if self.stop is not None and self.stop > num_ex:
# Warn once, then cap stop
logger.warning(
f"Stop parameter ({self.stop}) exceeds dataset size ({num_ex}). "
f"Using {num_ex} examples instead."
)
self.stop = num_ex
stop = self.num_examples() if self.stop is None else self.stop
stop = num_ex if self.stop is None else self.stop
step = self.step
span = stop - start
num = (span + step - 1) // step # ceil_div(span, step)