Custom Model for Learner Object,

[earldennison]

Hi, I was wondering if I could create a custom learner object, the model I would be dealing with is supposedly a spiked neural network. So basically create something like this:

learn = Learner(data=data, model=spiked_neural_network),

the custom model is made with pytorch. Is this possible?

Any help would be appreciated.

[averma (Abhinav Verma)]

Hi Earl, If you refer to the fast-ai nlp course. They’ve shown how you can add a custom GRU model or any other type of model to your learner object