mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-19 13:00:36 +08:00
* changed hard coded paramater, and moved it to parent_parser
```python
# ------------------------
# 4 INIT TRAINER
# ------------------------
trainer = Trainer(
experiment=exp,
checkpoint_callback=checkpoint,
early_stop_callback=early_stop,
gpus=hparams.gpus,
distributed_backend=hparams.dist_bak_end
)
parent_parser.add_argument('--dist_bak_end', type=str, default='ddp',
help='When using multiple GPUs set Trainer(distributed_backend=dp) (or ddp)')
```
* Update single_gpu_node_ddp_template.py