mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
pkg relative imports
* split requirements.txt * pytest verbose
This commit is contained in:
@@ -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