Preguntas con la etiqueta [tensor]

Lista tensor preguntas

¿Qué hace .view() en PyTorch?

9
respuestas
342
votos
259.7k
vistas

¿ Qué le hace .view()a un tensor x? ¿Qué significan los valores negativos? x = x.view(-1, 16 * 5 * 5)

¿Cómo guardo un modelo entrenado en PyTorch?

11
respuestas
382
votos
404.1k
vistas

¿Cómo guardo un modelo entrenado en PyTorch? He leído que: torch.save()/ torch.load()es para guardar/cargar un objeto serializable. model.state_dict()/ model.load_state_dict()es para guardar/cargar el estado del modelo.