mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-16 13:40:11 +08:00
14 lines
411 B
Python
14 lines
411 B
Python
import torch
|
|
|
|
from .test_module_base import LightningTestModelBase
|
|
from .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)
|