Right now, the main library lets users train their own custom Torch models by using the Trainer class. However, we want to give users who are not familiar with Torch the possibility to create some basic models.
Taking inspiration from the existing MLPworkflow, we first want to support the creation of LSTMs for both classification and generation tasks. The model should have a simple API and should internally use our Trainers (do not write any training-related code here).
The code should be located at fenn.nn.models.lstm
Please ask questions and collaborate with us while working on this. Feel free to join our Discord channel to collaborate with us!
Right now, the main library lets users train their own custom Torch models by using the
Trainerclass. However, we want to give users who are not familiar with Torch the possibility to create some basic models.Taking inspiration from the existing
MLPworkflow, we first want to support the creation of LSTMs for both classification and generation tasks. The model should have a simple API and should internally use our Trainers (do not write any training-related code here).The code should be located at
fenn.nn.models.lstmPlease ask questions and collaborate with us while working on this. Feel free to join our Discord channel to collaborate with us!