We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 82d853b commit 7013897Copy full SHA for 7013897
pytorch_lightning/trainer/ignored_warnings.py
@@ -3,12 +3,9 @@
3
4
def ignore_scalar_return_in_dp():
5
# Users get confused by this warning so we silence it
6
- m_1 = (
7
- 'Was asked to gather along dimension 0, but all '
8
- 'input tensors were scalars; will instead unsqueeze '
9
- 'and return a vector.'
10
- )
11
- warnings.filterwarnings('ignore', message=m_1)
+ warnings.filterwarnings('ignore', message='Was asked to gather along dimension 0, but all '
+ 'input tensors were scalars; will instead unsqueeze '
+ 'and return a vector.')
12
13
14
ignore_scalar_return_in_dp()
0 commit comments