OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Update parallel_apply.py for assertion error when len(modules) != len(inputs) (#94671)

Browse Source 
Print the result why it is wrong.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/94671
Approved by: https://github.com/ngimel, https://github.com/kit1980
This commit is contained in:
Kim,Won-Joong 2023-03-21 17:46:23 +00:00 committed by PyTorch MergeBot
parent a6bbeec2e1
commit c47cf9bc7f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/nn/parallel/parallel_apply.py
View File

@ -35,7 +35,7 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None):
element of :attr:`inputs` can either be a single object as the only argument
to a module, or a collection of positional arguments.
"""
assert len(modules) == len(inputs)
assert len(modules) == len(inputs), f'The number of modules {len(modules)} is not equal to the number of inputs {len(inputs)}'
if kwargs_tup is not None:
assert len(modules) == len(kwargs_tup)
else: