Views
No views yet
pytorch_model.bin: Checkpoint treinado do gerador.Gerador.py: Definição da arquitetura.config.json (opcional): Configuração básica do modelo.1from Gerador import Gerador
2import torch
3
4model = Gerador()
5model.load_state_dict(torch.load("pytorch_model.bin"))
6model.eval()
7
8# Imagem de exemplo
9input_tensor = torch.randn(1, 12, 18, 18) # (batch_size, canais, altura, largura)
10output = model(input_tensor)