5c3de2518362675e14b15f87044742a94b816ef2,snntoolbox/simulation/target_simulators/loihi_target_sim.py,SNN,add_layer,#SNN#Any#,104

Before Change

        // Convert weights to integers.
        if len(layer.weights):
            weights, biases = layer.get_weights()
            num_weight_bits = eval(self.config.get(
                "loihi", "connection_kwargs"))["numWeightBits"]
            weights, biases = to_integer(weights, biases, num_weight_bits)
            spike_layer.set_weights([weights, biases])

        self._previous_layer_name = layer.name

After Change

            weights, biases = layer.get_weights()
            weights, biases = to_integer(weights, biases, self.num_weight_bits)

            if "Flatten" in self._previous_layer_name:
                pl = self.parsed_model.get_layer(self._previous_layer_name)
                shape = pl.input_shape[1:]
                permutation = np.ravel(np.reshape(
                    np.arange(int(np.prod(shape))), shape, "F"), "C")
                weights = weights[permutation]

            spike_layer.set_weights([weights, biases])

        elif "AveragePooling" in get_type(layer):
            weights, biases = spike_layer.get_weights()

Instances