Hello,
I found a solution to the problem. This isn’t related to state_dict at all, but is rather just the string returned by a torch.Tensor being capped.
Use torch.set_printoptions(profile=“full”) to increase the limit.