Skip to content

Commit df36fba

Browse files
otiossSvetlana Karslioglu
and
Svetlana Karslioglu
authored
Fixes issue #1868 regarding TypeError: ZipperIterDataPipe instance doesn't have a valid length (#1954)
Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent ddcff31 commit df36fba

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

beginner_source/translation_transformer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ def train_epoch(model, optimizer):
309309
optimizer.step()
310310
losses += loss.item()
311311

312-
return losses / len(train_dataloader)
312+
return losses / len(list(train_dataloader))
313313

314314

315315
def evaluate(model):
@@ -333,7 +333,7 @@ def evaluate(model):
333333
loss = loss_fn(logits.reshape(-1, logits.shape[-1]), tgt_out.reshape(-1))
334334
losses += loss.item()
335335

336-
return losses / len(val_dataloader)
336+
return losses / len(list(val_dataloader))
337337

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

0 commit comments

Comments
 (0)