es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: GPYTORCH

Cómo utilizar nn.TransTransformerEncoder en PyTorch

Estoy tratando de utilizar el módulo “nn.TransformerEncoder” de PyTorch para una tarea de clasificación. Tengo puntos de datos de diferentes longitudes, es decir, tengo secuencias de diferentes longitudes. Todas las secuencias tienen una salida correspondiente (objetivo) que es 0 ó 1. [![enter code here][1]][1] Esta imagen muestra mi conjunto de . . . Read more

Guardar y cargar vocabulario y peso para utilizar torchtext en la inferencia.

He seguido este ejemplo para construir un clasificador de texto usando torchtext. Después de terminar el entrenamiento, he guardado el vocabulario y los pesos, pero cuando los he cargado de nuevo para hacer una inferencia con los datos de validación, obtuve cero de precisión. Encontré una pregunta similar en https://stackoverflow.com/questions/62744998/using-torchtext-for-inference . . . Read more

PyTorch se queda sin memoria de la GPU después de 1 época.

Entrené el modelo pero después de 1 epoch, ocurrió un error de CUDA. Cambié el tamaño de lote de entrenamiento a 1 y agregué torch.cuda.empty_cache(), pero nada cambió. ¿Cómo debo cambiarlo? “` Calentando el cargador de datos usando pin<em>memory en el dispositivo 0 Num por train<em>loader 562 Num por valid</em>loader . . . Read more