Skip to content

Commit b6a5804

Browse files
Ananth Subramaniamfacebook-github-bot
authored andcommitted
[PyTorchLightning/pytorch-lightning] remove deprecated train_loop (#10482)
Summary: ### New commit log messages fa0ed17f8 remove deprecated train_loop (#10482) Reviewed By: kandluis Differential Revision: D32454980 fbshipit-source-id: a35237dde06cc9ddac5373b75992ce88a6771c76
1 parent 2af7d2c commit b6a5804

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

mmf/trainers/lightning_core/loop_callback.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def _get_current_epoch_for_logging(self, trainer: Trainer):
160160
return trainer.current_epoch + 1
161161

162162
def _get_iterations_for_logging(self, trainer: Trainer):
163-
return trainer.train_loop.batch_idx + 1
163+
return trainer.fit_loop.batch_idx + 1
164164

165165
def _get_num_updates_for_logging(self, trainer: Trainer):
166166
return trainer.global_step + 1

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ datasets==1.2.1
1818
matplotlib==3.3.4
1919
pycocotools==2.0.2
2020
ftfy==5.8
21-
pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning@f9b9cdb0d
21+
pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning@fa0ed17f8
2222
torchaudio>=0.6.0, <=0.9.0
2323
psutil
2424
pillow==8.3.1

0 commit comments

Comments
 (0)