Obtener la salida del modelo mientras se utiliza PyTorch Lightning.
Estoy utilizando PyTorch Lightning para un proyecto. Mi modelo produce embeddings de palabras y quiero guardar las salidas del modelo en un archivo. ¿Cómo puedo acceder a las salidas del modelo después de que se haya completado el entrenamiento? ¿Debo usar callbacks? Mientras utilizo trainingendepoch(), cuando verifico train_outs, solo obtengo . . . Read more