mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
cleaned docs, fixed argparse generator (#1075)
* Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Test deprecated API for 0.8.0 and 0.9.0 (#1071) * till 0.8 * refactor * fix tests * fix tests * deprx till 0.9 * Update trainer.py * Apply suggestions from code review Co-authored-by: William Falcon <waf2107@columbia.edu> * updated test * updated test * updated test * updated test * updated test * updated test * updated test * updated test * updated test * updated test * updated test * updated test * updated test Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
This commit is contained in:
committed by
J. Borovec
co-authored by
Jirka Borovec
parent
9f140b7698
commit
be89eb07c8
@@ -19,6 +19,10 @@ modify the network. The `Trainer` can add all the available options to an Argume
|
||||
parser.add_argument('--layer_1_dim', type=int, default=128)
|
||||
parser.add_argument('--layer_2_dim', type=int, default=256)
|
||||
parser.add_argument('--batch_size', type=int, default=64)
|
||||
|
||||
# add all the available options to the trainer
|
||||
parser = pl.Trainer.add_argparse_args(parser)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
Now we can parametrize the LightningModule.
|
||||
|
||||
@@ -588,8 +588,12 @@ modify the network. The `Trainer` can add all the available options to an Argume
|
||||
|
||||
# parametrize the network
|
||||
parser.add_argument('--layer_1_dim', type=int, default=128)
|
||||
parser.add_argument('--layer_1_dim', type=int, default=256)
|
||||
parser.add_argument('--layer_2_dim', type=int, default=256)
|
||||
parser.add_argument('--batch_size', type=int, default=64)
|
||||
|
||||
# add all the available options to the trainer
|
||||
parser = pl.Trainer.add_argparse_args(parser)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
Now we can parametrize the LightningModule.
|
||||
|
||||
Reference in New Issue
Block a user