mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-11 12:31:23 +08:00
@@ -95,7 +95,7 @@ def test_tbd_remove_in_v1_0_0_model_hooks():
|
||||
|
||||
trainer = Trainer(logger=False)
|
||||
# TODO: why `dataloder` is required if it is not used
|
||||
result = trainer.evaluate(model, dataloaders=[[None]], max_batches=1)
|
||||
result = trainer._evaluate(model, dataloaders=[[None]], max_batches=1)
|
||||
assert result == {'val_loss': 0.6}
|
||||
|
||||
model = ModelVer0_7(hparams)
|
||||
@@ -106,5 +106,5 @@ def test_tbd_remove_in_v1_0_0_model_hooks():
|
||||
|
||||
trainer = Trainer(logger=False)
|
||||
# TODO: why `dataloder` is required if it is not used
|
||||
result = trainer.evaluate(model, dataloaders=[[None]], max_batches=1)
|
||||
result = trainer._evaluate(model, dataloaders=[[None]], max_batches=1)
|
||||
assert result == {'val_loss': 0.7}
|
||||
|
||||
Reference in New Issue
Block a user