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

adding framework level dp

Browse Source
This commit is contained in:
William Falcon committed 2019-06-25 18:09:29 -04:00
1 parent cbc627459a
commit 516f441153
2 files changed
+4 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

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