Skip to content

Commit fbe11fe

Browse files
committed
Fixes issue #1868 regarding TypeError: ZipperIterDataPipe instance doesn't have a valid length
1 parent faf2c47 commit fbe11fe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

beginner_source/translation_transformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ def train_epoch(model, optimizer):
303303
optimizer.step()
304304
losses += loss.item()
305305

306-
return losses / len(train_dataloader)
306+
return losses / len(list(train_dataloader))
307307

308308

309309
def evaluate(model):
@@ -327,7 +327,7 @@ def evaluate(model):
327327
loss = loss_fn(logits.reshape(-1, logits.shape[-1]), tgt_out.reshape(-1))
328328
losses += loss.item()
329329

330-
return losses / len(val_dataloader)
330+
return losses / len(list(val_dataloader))
331331

332332
######################################################################
333333
# Now we have all the ingredients to train our model. Let's do it!

0 commit comments

Comments
 (0)