mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
34c8c87f0f
commit
929980185d
1 changed files with 4 additions and 1 deletions
|
|
@ -152,7 +152,10 @@ class Run:
|
|||
f.write(str(train_time) + "\n")
|
||||
|
||||
def _validate(
|
||||
self, validator: Trainer, model: LightningModule, data_module: LightningDataModule
|
||||
self,
|
||||
validator: Trainer,
|
||||
model: LightningModule,
|
||||
data_module: LightningDataModule,
|
||||
) -> _EVALUATE_OUTPUT:
|
||||
score = validator.validate(model, datamodule=data_module)
|
||||
return score
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue