File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
src/lightning/pytorch/plugins/precision Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 1111# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212# See the License for the specific language governing permissions and
1313# limitations under the License.
14- from collections .abc import Generator
14+ from collections .abc import Generator , Iterable
1515from contextlib import AbstractContextManager , contextmanager
16- from typing import Any , Literal , Iterable
16+ from typing import Any , Literal
1717
1818import torch
1919import torch .nn as nn
@@ -71,6 +71,7 @@ class LightningDoublePrecisionModule(_DeviceDtypeModuleMixin, nn.Module):
7171 pl_module: the model to wrap
7272
7373 """
74+
7475 _ddp_params_and_buffers_to_ignore : Iterable [str ]
7576
7677 def __init__ (self , pl_module : "pl.LightningModule" ) -> None :
You can’t perform that action at this time.
0 commit comments