Custom Model for Learner Object,

(Earl Dennison) #1

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.

0 Likes