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: PYTORCH-LIGHTNING

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

PyTorch Lightning: la salida del entrenamiento de la CNN con valores complejos arroja NaN después de 1 lote.

He construido una CNN con valores complejos utilizando ComplexPyTorch, donde las capas están envueltas en un torch.ModuleList. Cuando ejecuto la red, pasa correctamente la comprobación de validación y 1 lote del entrenamiento, pero luego mi pérdida devuelve NaNs. Al registrar los gradientes en on_after_backward, se muestran NaNs inmediatamente. ¿Alguien tiene . . . Read more