"""第 18 课补全练习：保存并还原参数字典。修改 solve，保持下方检查不变。"""
import math, io, argparse
import torch
from torch import nn
from torch.nn import functional as F
torch.set_num_threads(2)
torch.manual_seed(26)

def solve(state):
    buffer=io.BytesIO()
    torch.save(state,buffer)
    buffer.seek(0)
    return torch.load(buffer,weights_only=True)

source={'weight':torch.tensor([1.,2.])}
assert torch.equal(solve(source)['weight'],source['weight'])
print("本课补全练习通过。")