mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-21 13:20:08 +08:00
14 lines
417 B
Python
14 lines
417 B
Python
import torch
|
|
|
|
from .lm_test_module_base import LightningTestModelBase
|
|
from .lm_test_module_mixins import LightningValidationMixin, LightningTestMixin
|
|
|
|
|
|
class LightningTestModel(LightningValidationMixin, LightningTestMixin, LightningTestModelBase):
|
|
"""
|
|
Most common test case. Validation and test dataloaders
|
|
"""
|
|
|
|
def on_training_metrics(self, logs):
|
|
logs['some_tensor_to_test'] = torch.rand(1)
|