1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

Tạo một khối tuần tự

Bạn quyết định thiết kế lại mẫu mô hình CNN nhị phân của mình bằng cách tạo một khối các tầng tích chập. Cách này giúp bạn xếp chồng nhiều tầng theo thứ tự. Với mô hình cải tiến này, bạn sẽ dễ dàng thiết kế nhiều kiến trúc CNN khác nhau.

torch và torch.nn dưới tên nn đã được nhập.

Hướng dẫn

100 XP
  • Trong phương thức __init__(), định nghĩa một khối các tầng tích chập và gán cho self.conv_block.
  • Trong lượt truyền forward(), đưa đầu vào đi qua khối tích chập bạn đã định nghĩa.