Preguntas con la etiqueta [pytorch]

Lista pytorch preguntas

¿Qué hace .contiguous() en PyTorch?

8
respuestas
215
votos
145.4k
vistas

¿ Qué hace x.contiguous()por un tensor x?

¿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)

En la implementación MultiHeadAttention de Pytorch , con respecto a in_proj_weight, ¿es cierto que los primeros elementos embed_dim corresponden a la consulta, los siguientes elementos embed_dim corresponden a la clave

¿Cómo imprimo el resumen del modelo en PyTorch?

11
respuestas
337
votos
415.3k
vistas

¿Cómo imprimo el resumen de un modelo en PyTorch como lo model.summary()hace en Keras? Model Summary: ____________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ==================================================================================================== input_1 (InputLayer) (None, 1,

¿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.