Lightning-AI/pytorch-lightning
trainer.test() with given checkpoint logs last epoch instead of checkpoint epoch
オープン
#20,052 opened on 2024/07/05
bughelp wantedrepro needed
Repository metrics
- Stars
- (26,687 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
Bug description
Testing from a given checkpoint leads to logging the epoch number of the last checkpoint instead of the checkpoint specified:
trainer = Trainer(..., max_epochs=10)
lightning_module = MyLightningModule(...)
datamodule = MyDatamodule()
trainer.fit(lightning_module , datamodule=datamodule)
trainer.test(lightning_module , datamodule=datamodule, ckpt_path="last") # <-- ok: logs correct epoch and step
ckpt_path="/.../checkpoints/epoch=2-step=396.ckpt"
trainer.test(lightning_module , datamodule=datamodule, ckpt_path=ckpt_path) # <-- incorrect: logs last epoch and step
The second test logs epoch 10 instead of epoch 2. Similarly, the step number of the second test is incorrect.
What version are you seeing the problem on?
v2.2.1