mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
William Falcon
8b2a2aeda3
* added ignore warnings module * added ignore warnings module * Fixes #249 * Update ignored_warnings.py
15 lines
345 B
Python
15 lines
345 B
Python
import warnings
|
|
|
|
|
|
def ignore_scalar_return_in_dp():
|
|
# Users get confused by this warning so we silence it
|
|
m_1 = """
|
|
Was asked to gather along dimension 0, but all
|
|
input tensors were scalars; will instead unsqueeze
|
|
and return a vector.
|
|
"""
|
|
warnings.filterwarnings('ignore', message=m_1)
|
|
|
|
|
|
ignore_scalar_return_in_dp()
|