# Codeblock 1
import torch
import torch.nn as nn
from torchinfo import summary