Performance, Pytorch, Deep Learning, Artificial Intelligence, Tensorboard.
torch.utils.data from torch import Tensor class Net(nn.Module): def __init__(self, num_hidden=10, num_classes=10): super().__init__() self.conv_in = nn.Conv2d(3, 10, 3, padding='same') hidden = [] for i in range(num_hidden): hidden.append(nn.Conv2d(10, 10, 3, padding='same'))