Skip to content

Commit 0ff985b

Browse files
Fix spelling
1 parent 5c795d0 commit 0ff985b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recipes_source/recipes/module_load_state_dict_tips.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def my_processing_function(key, device):
158158

159159
new_m.load_state_dict(state_dict, assign=True)
160160
# Before 2.3.0, this MUST be done AFTER the load_state_dict with assign.
161-
# In versions >= 2.3.0, one can consider setting torch.__future__.set_swap_module_params_on_conversion
161+
# In versions >= 2.3.0, one can consider setting ``torch.__future__.set_swap_module_params_on_conversion``
162162
opt = torch.optim.SGD(new_m.parameters(), lr=1e-3)
163163

164164
###############################################################################

0 commit comments

Comments
 (0)