Skip to content

Commit

Permalink
[bugfix] Fix signature mismatch in DDPCPUHPCAccelerator's model_to_de…
Browse files Browse the repository at this point in the history
…vice (#5505)

* Update ddp_cpu_hpc_accelerator.py

* Update CHANGELOG.md
  • Loading branch information
ananthsub authored Jan 16, 2021
1 parent 7f352cb commit 6926b84
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
- Logging only on `not should_accumulate()` during training ([#5417](https://github.com/PyTorchLightning/pytorch-lightning/pull/5417))
- Resolve interpolation bug with Hydra ([#5406](https://github.com/PyTorchLightning/pytorch-lightning/pull/5406))
- Check environ before selecting a seed to prevent warning message ([#4743](https://github.com/PyTorchLightning/pytorch-lightning/pull/4743))
- Fixed signature mismatch in `model_to_device` of `DDPCPUHPCAccelerator` ([#5505](https://github.com/PyTorchLightning/pytorch-lightning/pull/5505))


## [1.1.3] - 2021-01-05
Expand Down
2 changes: 1 addition & 1 deletion pytorch_lightning/accelerators/ddp_cpu_hpc_accelerator.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def __init__(self,
super().__init__(trainer, cluster_environment, ddp_plugin)
self.nickname = 'ddp_cpu'

def model_to_device(self, model, process_idx):
def model_to_device(self, model):
model.cpu()

def get_device_ids(self):
Expand Down

0 comments on commit 6926b84

Please sign in to comment.