قبول نموذج اصطناعي كامل

import copy
import torch
from torch import nn
class Fixture(nn.Module):
    def __init__(self):
        super().__init__()
        self.register_buffer("center",torch.tensor([10.0,20.0],dtype=torch.float64,device="cpu"))
        self.register_buffer("scale",torch.tensor([2.0,4.0],dtype=torch.float64,device="cpu"))
        self.body = nn.Sequential(nn.Linear(2,2,bias=False,dtype=torch.float64,device="cpu"),nn.ReLU(),
                                  nn.Linear(2,1,bias=False,dtype=torch.float64,device="cpu"))
        with torch.no_grad():
            self.body[0].weight.copy_(torch.eye(2,dtype=torch.float64,device="cpu"))
            self.body[2].weight.copy_(torch.tensor([[1.0,-1.0]],dtype=torch.float64,device="cpu"))
    def forward(self,x):
        return self.body((x-self.center)/self.scale)
model = Fixture().eval()
restored = Fixture()
restored.load_state_dict(copy.deepcopy(model.state_dict()),strict=True)
restored.eval()
x = torch.tensor([[12.0,24.0],[14.0,20.0]],dtype=torch.float64,device="cpu")
with torch.inference_mode():
    output = model(x)
    repeated = restored(x)
print("shape", tuple(output.shape))
print("predictions", output.tolist())
print("same restored", torch.equal(output,repeated))

shape (2, 1)
predictions [[0.0], [2.0]]
same restored True

تعرف مخازن التحويل المسجلة والأوزان المسندة مثال استدلال يمكن فحصه بالكامل؛ ولم تنفذ Module أو استرجاع حالة أو inference_mode أو مخرجات نموذج.

from math import isfinite
def score(values,units):
    if units != ["cm","cm"]:
        raise ValueError("expected centimeters")
    if len(values)!=2 or not all(type(x) in (int,float) and isfinite(x) for x in values):
        raise ValueError("expected two finite measurements")
    z = [(x-m)/s for x,m,s in zip(values,[10.0,20.0],[2.0,4.0])]
    hidden = [max(0.0,x) for x in z]
    return hidden[0]-hidden[1]
print("scores", [score(row,["cm","cm"]) for row in [[12.0,24.0],[14.0,20.0]]])
try:
    score([12.0,24.0],["mm","mm"])
except ValueError as error:
    print(str(error))

scores [0.0, 2.0]
expected centimeters

يفحص المرافق المنفذ سياسة الوحدات المسماة والمدخل المنتهي والتطبيع والتقويم وحساب الدرجة. ولا يثبت قبول إطار أو جودة تنبؤية واقعية.