Je veux que pytorch dessine le code de la courbe de perte

Je peux offrir quelques conseils pour vous aider à tracer la courbe de perte de PyTorch. Tout d'abord, vous pouvez calculer la valeur de perte à l'aide de la fonction nn.BCEWithLogitsLoss() de PyTorch. Vous pouvez ensuite utiliser la fonction optim.SGD() de PyTorch pour initialiser les paramètres du modèle et la fonction nn.Module.fit() de PyTorch pour former le modèle. Enfin, vous pouvez utiliser la fonction torch.Tensor.plot() de PyTorch pour tracer la courbe de perte.

Je suppose que tu aimes

Origine blog.csdn.net/weixin_35755562/article/details/129523408
conseillé
Classement