snwagh / falcon-public

Implementation of protocols in Falcon
90 stars 46 forks source link

Save the model after training and make prediction #50

Open IrinaKarmatskikh opened 1 year ago

IrinaKarmatskikh commented 1 year ago

Hello, I would like to train a neural network using a falcon and then make a prediction. Is this possible with the current functionality? Function "test" loads a pre-trained model, can I save the model after training with falcon?