AI — PyTorch

What is PyTorch?

Open-source deep learning framework by Facebook.

Install

pip install torch torchvision

Basic Model

import torch
import torch.nn as nn
import torch.optim as optim

class SimpleNet(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc1 = nn.Linear(784, 128)
        self.fc2 = nn.Linear(128, 10)
    
    def forward(self, x):
        x = torch.relu(self.fc1(x))
        x = self.fc2(x)
        return x

model = SimpleNet()
criterion = nn.CrossEntropyLoss()
optimizer = optim.Adam(model.parameters())

Training Loop

for epoch in range(10):
    for batch in dataloader:
        optimizer.zero_grad()
        output = model(batch.data)
        loss = criterion(output, batch.target)
        loss.backward()
        optimizer.step()

Mini Practice

1. Install PyTorch
2. Build neural network
3. Train model
4. Compare with TensorFlow

Up Next

Continue with OpenAI — OpenAI API.