diff --git a/pytorch_lightning/core/__init__.py b/pytorch_lightning/core/__init__.py index 56d6a258..8a422344 100644 --- a/pytorch_lightning/core/__init__.py +++ b/pytorch_lightning/core/__init__.py @@ -94,157 +94,6 @@ Check out this `COLAB `_ -# is available anywhere in your model by calling self.hparams. -# -# **Return** -# An argument parser -# -# **Example** -# -# .. code-block:: python -# -# @staticmethod -# def add_model_specific_args(parent_parser, root_dir): -# parser = HyperOptArgumentParser(strategy=parent_parser.strategy, parents=[parent_parser]) -# -# # param overwrites -# # parser.set_defaults(gradient_clip_val=5.0) -# -# # network params -# parser.opt_list('--drop_prob', default=0.2, options=[0.2, 0.5], type=float, tunable=False) -# parser.add_argument('--in_features', default=28*28) -# parser.add_argument('--out_features', default=10) -# # use 500 for CPU, 50000 for GPU to see speed difference -# parser.add_argument('--hidden_dim', default=50000) -# -# # data -# parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str) -# -# # training params (opt) -# parser.opt_list('--learning_rate', default=0.001, type=float, -# options=[0.0001, 0.0005, 0.001, 0.005], tunable=False) -# parser.opt_list('--batch_size', default=256, type=int, -# options=[32, 64, 128, 256], tunable=False) -# parser.opt_list('--optimizer_name', default='adam', type=str, -# options=['adam'], tunable=False) -# return parser -# -# """ - from .lightning import LightningModule __all__ = ['LightningModule']