Set initial value for membrane potential

[JackCaster]

Set initial value for membrane potential

[Met4physics]

You can write a function which make a traversal in your model to set initial membrane potential. For example:

def set_initial(model, value):
    for module in model.modules():
        if isinstance(module, neuron):
            module.v = value

[JackCaster]

You can write a function which make a traversal in your model to set initial membrane potential. For example:

def set_initial(model, value):
    for module in model.modules():
        if isinstance(module, neuron):
            module.v = value

Thanks, this is close to what I was looking for! But I was wondering if it would be possible to have the initial value being a parameter to learn

[Met4physics]

You can write a function which make a traversal in your model to set initial membrane potential. For example:

def set_initial(model, value):

for module in model.modules():

    if isinstance(module, neuron):

        module.v = value

Thanks, this is close to what I was looking for! But I was wondering if it would be possible to have the initial value being a parameter to learn

Then you need to use nn.parameter to represent the initial value.

[JackCaster]

Yes exactly, I think I need a parameter v_init that by default is set to 0.0. Where is the right place to add this parameter. I usually subclass the base neuron to add these kind of changes, but I struggle to find the right way to add this parameter in there.

[Met4physics]

Yes exactly, I think I need a parameter v_init that by default is set to 0.0. Where is the right place to add this parameter. I usually subclass the base neuron to add these kind of changes, but I struggle to find the right way to add this parameter in there.

Just like this:

class Neuron():
    def __init__(self):
        self.init_v = nn.parameter([1])

Note that this code is only a draft for illustration. And you should also write a new reset function to set v to be equal with v_init after a timestep.