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

Reentrenar un modelo BERT

He entrenado un modelo BERT utilizando pytorch con alrededor de un millón de datos de texto para una tarea de clasificación. Después de probar este modelo con nuevos datos, obtengo Falsos Positivos y Falsos Negativos. Ahora quiero volver a entrenar el modelo existente solo con los FN y FP. No . . . Read more

Función de activación por tramos.

Estoy tratando de escribir una función de activación por partes cuya pendiente entre -6 y 0 es 0.1 y en los demás lugares es uno. Y el tamaño de entrada (X) es (B, C, H, W). Así que llegué a la conclusión de que la mejor manera es el código . . . Read more

CUDA OOM – ¿Pero los números no cuadran?

Estoy tratando de entrenar un modelo utilizando PyTorch. Al comenzar el entrenamiento del modelo, obtengo el siguiente mensaje de error: “RuntimeError: CUDA sin memoria. Se intentó asignar 5.37 GiB (GPU 0; capacidad total de 7.79 GiB; 742.54 MiB ya asignados; 5.13 GiB libres; 792.00 MiB reservados en total por PyTorch)” . . . Read more