Pytorch skill

O que ela faz

A skill é um guia de boas práticas que o Claude segue ao escrever código PyTorch: modelos montados em classes nn.Module, inferência sem cálculo de gradiente, pesos inicializados direito e acúmulo de gradiente sem erro. É texto de orientação, sem script próprio.

Cobre dois tipos de modelo: transformers, com a biblioteca da Hugging Face e ajuste fino mais leve via LoRA ou P-tuning, e modelos de difusão de imagem, com a biblioteca Diffusers, agendadores de ruído e variantes como o SDXL.

No treino, pede divisão em treino, validação e teste, agenda na taxa de aprendizado, parada antecipada, corte de gradiente pra dar estabilidade e cuidado com valores NaN. Pra ganhar velocidade, usa precisão mista, várias GPUs e medição pra achar o gargalo. No fim, dá pra montar uma demo no Gradio pra testar o modelo no navegador.

Nas palavras de quem fez

Quando usar

• Fazer ajuste fino de um modelo da Hugging Face com LoRA nos seus dados
• Montar um pipeline de geração de imagem com Diffusers e SDXL
• Acelerar um treino que está lento ou estourando a memória da GPU
• Criar uma demo no Gradio pra mostrar o modelo pro cliente
• Achar de onde vem o NaN que aparece no meio do treino

Depois de instalar, é só pedir assim

• Escreve um script em PyTorch pra fazer ajuste fino com LoRA nesse modelo da Hugging Face usando o meu CSV de atendimento
• Meu treino dá NaN depois de algumas épocas, revisa o código e corrige
• Monta uma demo no Gradio que gera imagem com SDXL a partir de um texto

Segurança

Gen Agent Trust Hub, Socket e Snyk leem o código da skill atrás de comando perigoso, dado vazando e instrução escondida. Alerta não é reprovação: quer dizer que vale ler o que a skill faz antes de instalar.

• Gen Agent Trust HubPassou
• SocketPassou
• SnykPassou

Perguntas