Introducción

Este es mi primer post. Probando syntax highlighting:

1import torch
2import numpy as np
3
4def train_model(data):
5    model = torch.nn.Linear(10, 1)
6    optimizer = torch.optim.Adam(model.parameters())
7    return model