wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

adding framework level dp

Browse Source
This commit is contained in:
William Falcon
2019-06-25 18:09:29 -04:00
parent cbc627459a
commit 516f441153
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/fully_featured_trainer.py
+1 -1
View File
@@ -164,7 +164,7 @@ if __name__ == '__main__':
# allow model to overwrite or extend args
TRAINING_MODEL = AVAILABLE_MODELS[model_name]
parser = TRAINING_MODEL.add_model_specific_args(parent_parser)
parser = TRAINING_MODEL.add_model_specific_args(parent_parser, root_dir)
hyperparams = parser.parse_args()
# format GPU layout