Thanks for the response,  how would you interpret the first two layers based on the top picture? Is 64 the input and output size? is 5x5 the kernal size?
My guess so far would be:
class TargetA(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(? , ?, ?)
        self.conv2 = nn.Conv2d(?, ?, ?)
        self.drop1 = nn.Dropout(0.25)
        self.fc1 = nn.Linear(,128)
        self.drop2 = nn.Dropout(0.5)
        self.fc2 = nn.Linear(128,10)