device = 'cuda' if torch.cuda.is_available() else 'cpu' ... val_losses.append(val_loss.item())print(model.state_dict())
確定! 回上一頁