red2=torchvision.models.mnasnet0_5(weights=torchvision.models.MNASNet0_5_Weights.DEFAULT) #La parte final es classifier red2.classifier=nn.Sequential(nn.Linear(1280,101),nn.Sigmoid()) # Cualquier cosa, pero entran 1280 y salen 101 #La parte inicial es layers for p in red2.layers.parameters(): p.requires_grad=False