mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
pkg relative imports
* split requirements.txt * pytest verbose
This commit is contained in:
1 parent
92f8c57ff5
commit
469941a528
20 files changed
+54
-59
No files matched your search
@@ -1,9 +1,10 @@
|
||||
import torch
|
||||
from pytorch_lightning.root_module.memory import ModelSummary
|
||||
from pytorch_lightning.root_module.grads import GradInformation
|
||||
from pytorch_lightning.root_module.model_saving import ModelIO, load_hparams_from_tags_csv
|
||||
from pytorch_lightning.root_module.hooks import ModelHooks
|
||||
from pytorch_lightning.root_module.decorators import data_loader
|
||||
|
||||
from .memory import ModelSummary
|
||||
from .grads import GradInformation
|
||||
from .model_saving import ModelIO, load_hparams_from_tags_csv
|
||||
from .hooks import ModelHooks
|
||||
from .decorators import data_loader
|
||||
|
||||
|
||||
class LightningModule(GradInformation, ModelIO, ModelHooks):
|
||||
|
||||
Reference in new issue
Block a user