Torch state_dict

Creator

Creator

Seonglae Cho

Created

Created

2025 Apr 10 20:12

Editor

Editor

Seonglae Cho

Edited

Edited

2025 Apr 10 20:13

Refs

Refs

learnable parameters of a `torch.nn.Module`

A state_dict is simply a Python dictionary object that maps each layer to its parameter tensor in model.parameters().

What is a state_dict in PyTorch — PyTorch Tutorials 2.6.0+cu124 documentation

Created On: Apr 17, 2020 | Last Updated: Feb 06, 2024 | Last Verified: Nov 05, 2024

https://pytorch.org/tutorials/recipes/recipes/what_is_state_dict.html

Backlinks

torch.nn.Module register_buffer()torch.save()

Recommendations

//////////