Xây dựng mạng nơ-ron tích chập đơn giản để nhận diện hình ảnh CIFAR10 bằng PyTorch
Thiết lập:
python 3.11.1
pytorch 2.3.0
Chuẩn bị ban đầu
1. Cấu hình GPU
import torch
import torch.nn as nn
import matplotlib.pyplot as plt
import torchvision
thiet_bi = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(thiet_bi)
2. Nhập dữ liệu
Sử dụng dataset tải xuống tập dữ liệu CIFAR10 và phân chia ...
Đăng vào ngày 16 tháng 6 lúc 08:21