File tree Expand file tree Collapse file tree 2 files changed +2
-0
lines changed Expand file tree Collapse file tree 2 files changed +2
-0
lines changed Original file line number Diff line number Diff line change 2
2
3
3
Tips for Loading an ``nn.Module`` from a Checkpoint
4
4
===================================================
5
+ **Author:** `Mikayla Gawarecki <https://github.com/mikaylagawarecki>`_
5
6
6
7
If you're loading a checkpoint and want to reduce compute and memory as much as possible,
7
8
this tutorial shares some recommended practices. In particular, we will discuss
Original file line number Diff line number Diff line change 1
1
"""
2
2
Extension points in ``nn.Module`` for ``load_state_dict`` and tensor subclasses
3
3
===============================================================================
4
+ **Author:** `Mikayla Gawarecki <https://github.com/mikaylagawarecki>`_
4
5
5
6
This recipe introduces a new utility function ``torch.utils.swap_tensors``
6
7
as well as two new extension points where it has been integrated in
You can’t perform that action at this time.
0 commit comments