add metrics to pl module (#2506)
This commit is contained in:
parent
9924c76faa
commit
b6507daf89
|
@ -54,6 +54,7 @@ else:
|
|||
from pytorch_lightning.callbacks import Callback
|
||||
from pytorch_lightning.trainer import Trainer
|
||||
from pytorch_lightning.utilities.seed import seed_everything
|
||||
from pytorch_lightning import metrics
|
||||
|
||||
__all__ = [
|
||||
'Trainer',
|
||||
|
@ -61,6 +62,7 @@ else:
|
|||
'Callback',
|
||||
'data_loader',
|
||||
'seed_everything',
|
||||
'metrics'
|
||||
]
|
||||
|
||||
# necessary for regular bolts imports. Skip exception since bolts is not always installed
|
||||
|
|
Loading…
Reference in New Issue