示例代碼

常用的代碼示例和最佳實踐。

PyTorch 訓練示例

import torch
import torch.nn as nn

# 檢查 GPU
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(f'Using device: {device}')

# 定義模型
model = nn.Sequential(
    nn.Linear(784, 256),
    nn.ReLU(),
    nn.Linear(256, 10)
).to(device)