Skip to content

Commit 9e99865

Browse files
author
Azure Pipelines
committed
Merge remote-tracking branch 'origin/main' into publication
2 parents 57e2a2a + 2887239 commit 9e99865

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

lightning_examples/text-transformers/.meta.yml

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
title: Finetune Transformers Models with PyTorch Lightning
22
author: PL team
33
created: 2021-01-31
4-
updated: 2021-06-21
4+
updated: 2021-12-03
55
license: CC BY-SA
6-
build: 1
6+
build: 2
77
tags:
88
- Text
99
description: |
@@ -15,6 +15,7 @@ requirements:
1515
- datasets
1616
- scipy
1717
- scikit-learn
18+
- torchtext>=0.9
1819
accelerator:
1920
- CPU
2021
- GPU

lightning_examples/text-transformers/text-transformers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ def setup(self, stage=None) -> None:
220220
if stage != "fit":
221221
return
222222
# Get dataloader by calling it - train_dataloader() is called after setup() by default
223-
train_loader = self.train_dataloader()
223+
train_loader = self.trainer.datamodule.train_dataloader()
224224

225225
# Calculate total steps
226226
tb_size = self.hparams.train_batch_size * max(1, self.trainer.gpus)

0 commit comments

Comments
 (0)