add metrics to pl module (#2506)

This commit is contained in:
davinnovation 2020-07-05 12:55:22 +09:00 committed by GitHub
parent 9924c76faa
commit b6507daf89
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -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