你即將離開本站

並前往https://pytorch-lightning.readthedocs.io/en/latest/common/lightning_module.html

LightningModule — PyTorch Lightning 1.6.0dev documentation

def training_step(self, batch, batch_idx): x, y = batch y_hat = self.model(x) loss = F.cross_entropy(y_hat, y) # logs metrics for each training_step, ...

確定！ 回上一頁

查詢 「PyTorch loss」的人也找了：

  1. PyTorch loss function
    2.
  2. --Pytorch自 訂 loss function
    3.
  3. CrossEntropyLoss
    4.
  4. Pytorch Dense
    5.
  5. Pytorch leaky ReLU
    6.
  6. Pytorch loss backward
    7.
  7. Pytorch dropout
    8.
  8. Pytorch softmax
    9.