val_loss.append(loss). val_loss = torch.mean(torch.tensor(val_loss)). model.train(). Lightning code. # model. class LITAutoEncoder(pl.LightningModule):.
確定! 回上一頁